【问题标题】:cudnn can't be installcudnn 无法安装
【发布时间】:2017-04-12 04:25:04
【问题描述】:

我正在使用 cuda 8.0 windows 10 GTX 770M 在 theano 上尝试LeNet5,似乎无法正确更新参数。 当我导入theano时,我会得到这个错误:

Can not use cuDNN on context None: cannot compile with cuDNN. We got this error:
c:\users\yanjun~1\appdata\local\temp\try_flags_8n7fhy.c:4:19: fatal error: cudnn.h: No such file or directory
compilation terminated.

Mapped name None to device cuda: GeForce GTX 770M (0000:01:00.0)

我已通过将文件复制到“NVIDIA GPU Computing Toolkit\CUDA\v8.0”将 cudnn 5 安装到我的 cuda 8.0,但仍然显示相同的错误。

是不是因为theano不能编译cudnn,conv2dpool_2d都不能正常工作?结果我训练的时候错误率一直保持在90%左右。我是否正确安装了cudnn?我只是关注 theano 文档中的instructions,但 zip 中没有 .so* 文件。

我试过cpu模式,可以正常训练但是真的很慢。

希望你们能帮助我摆脱困境。

非常感谢!

【问题讨论】:

  • 你的问题解决了吗?

标签: theano theano-cuda cudnn


【解决方案1】:

我在迁移到新的 theano 后端时遇到了类似的问题,并通过以下步骤修复了它(在我的 conda 环境中)。

  1. 使用conda update theano 更新theano 并使用conda install -c rdonnelly pygpu 获取pygpy。我正在安装一个 RC 包,因为它提供的问题更少。
  2. 检查是否需要更改代码here 因为我使用的是 keras,所以我不必这样做。
  3. 将 theanorc 从 device = gpu 更改为 device = cuda0。如果您访问了上述链接,您可能已经这样做了。
  4. 添加到您的 theanorc 文件的底部:

    compiler_bindir = C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\bin
    
    [cuda]
    root = C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
    
    [dnn]
    library_path = C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
    include_path = C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include
    

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2017-02-10
    • 1970-01-01
    • 2017-10-25
    • 2015-09-28
    • 1970-01-01
    相关资源
    最近更新 更多