【问题标题】:nvcc fatal : Value 'sm_61' is not defined for option 'gpu-architecture' error with theanonvcc 致命:没有为选项“gpu-architecture”定义值“sm_61”,theano 错误
【发布时间】:2016-11-02 16:18:09
【问题描述】:

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

【问题讨论】:

  • 您需要切换到支持该特定编译选项的 CUDA 8.0RC,否则将 theano 配置更改为不针对您的 GTX 1080 GPU 类型进行编译。
  • 您能否详细说明第二个选项?

标签: python cuda gpu theano nvcc


【解决方案1】:

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

【讨论】:

    【解决方案2】:

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

    flags = -arch=sm_52
    

    不再有 nvcc 致命错误

    【讨论】:

      猜你喜欢
      • 2018-05-19
      • 1970-01-01
      • 1970-01-01
      • 2016-05-23
      • 1970-01-01
      • 2019-08-02
      • 2021-09-08
      • 2021-02-14
      • 2014-07-27
      相关资源
      最近更新 更多