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

时间:2016-06-30 13:56:55

标签: python cuda gpu theano nvcc

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

2 个答案:

答案 0 :(得分:7)

正如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不是最终版本。

答案 1 :(得分:2)

我能够通过在.theanorc文件中包含以下行来找到这个问题的解决方案(因为我仍然想使用CUDA 7.5)

flags = -arch=sm_52

没有更多的nvcc致命错误