cudaCheckError() failed : invalid device function解决方法_服务器

先将出现的问题截图,上图所示为我遇见的错误。

我在服务器上(ubuntu14.04),运行smallcorgi/Faster-RCNN的tensorflow代码时候出现的问题,我的显卡是Tesla K40。

在百度后发现根本没有切实可行的方法,最后还是使用Google解决了问题,发现在运行代码时候在lib文件夹下面执行make操作的时候需要将make.sh文件进行修改,将arch参数从sm_37改为sm_35。为大家附上参数列表。

cudaCheckError() failed : invalid device function解决方法_tensorflow_02