nvcc fatal : Value 'sm_61' is not defined for option 'gpu-architecture' error with theano

nvcc fatal : Value 'sm_61' is not defined for option 'gpu-architecture' error with theano

我正在设置 python 和 theano 以便与 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 不是最终版本。

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

flags = -arch=sm_52

不再有 nvcc 致命错误