我正在设置
python和theano用于gpu on;
ubuntu 14.04,
GeForce GTX 1080
已成功为系统安装了NVIDIA驱动程序(367.27)和CUDA工具包(7.5),
但在使用theano gpu实现进行测试时,我得到了上述错误(例如;在启用gpu的情况下导入theano时)
我试图寻找可能的解决方案,但没有成功.
我对ubuntu和gpu编程有点新意,所以我很感激能够解决这个问题.
谢谢
ubuntu 14.04,
GeForce GTX 1080
已成功为系统安装了NVIDIA驱动程序(367.27)和CUDA工具包(7.5),
但在使用theano gpu实现进行测试时,我得到了上述错误(例如;在启用gpu的情况下导入theano时)
我试图寻找可能的解决方案,但没有成功.
我对ubuntu和gpu编程有点新意,所以我很感激能够解决这个问题.
谢谢
解决方法
正如Robert Crovella所说,SM 6.1(sm_61)仅在CUDA 8.0及更高版本中受支持,因此您应该从
https://developer.nvidia.com/cuda-toolkit下载CUDA 8.0 Release Candidate
支持Ubuntu 14.04,网站上有关如何设置的说明应该很简单(将行复制并粘贴到控制台).
我还建议下载CUDA 8.0,因为RC不是最终版本.