nvcc致命:没有为theano选项'gpu-architecture'错误定义值'sm_61'

Question

我正在设置python和theano用于gpu on; ubuntu 14.04,GeForce GTX 1080已经成功为系统安装了NVIDIA驱动程序(367.27)和CUDA工具包(7.5),但是在使用theano gpu实现进行测试时我得到了上述错误(例如;当导入带有gpu的theano时)我试过了寻找可能的解决方案,但没有成功.我对ubuntu和gpu编程有点新意,所以我很感激能够解决这个问题.谢谢

Answer 1

正如Robert Crovella所说,SM 6.1(sm_61)仅在CUDA 8.0及更高版本中受支持,因此您应该从https://developer.nvidia.com/cuda-toolkit下载CUDA 8.0候选版本.

支持Ubuntu 14.04,网站上有关如何设置的说明应该很简单(将行复制并粘贴到控制台).

我还建议下载CUDA 8.0,因为RC不是最终版本.

Answer 2

通过在 .theanorc 文件中包含以下行，我能够找到此问题的解决方案（因为我仍然想使用 CUDA 7.5）

flags = -arch=sm_52

不再有 nvcc 致命错误