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