2017-04-12 134 views
1

我想在theano上使用cuda 8.0 windows 10 GTX 770M上的LeNet5,而且看起来params无法正确更新。通过复制文件到“NVIDIA GPU计算工具包\ CUDA \ V8.0”cudnn无法安装

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) 

我已经安装了cudnn 5到我的CUDA 8.0,但仍然是相同的: 当我输入theano,我会得到这个错误错误显示。

是否因为theano无法编译cudnn,conv2dpool_2d都无法正常工作?因此,我的错误率在训练时保持在90%左右。我正确安装了cudnn吗?我只是按照theano的文档上的instructions,但在zip中没有.so *文件。

我试过cpu模式,它可以被正确的训练,但真的很慢。

希望你们能帮助我摆脱这种困扰我的日子。

非常感谢!

+0

你有没有解决问题了吗? – jingweimo

回答

0

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

  1. 更新theano与conda update theano并获得pygpy与conda install -c rdonnelly pygpu。我正在安装一个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