我正在使用Cuda8.0 windows 10 GTX770m在theano上试用LeNet5,而且似乎不能正确地更新params。当我导入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计算工具包\ cuda \v8.0“,我已经将cudnn 5安装到我的CUDA 8.0上,但仍然显示了相同的错误。
是不是因为theano不能编译cudnn,conv2d和pool_2d都不能正常工作?结果,我训练时的错误率保持在90%左右。我安装的cudnn正确吗?我只是跟踪了theano文档上的使用说明,但在zip中没有.so*文件。
我试过cpu模式,它可以被适当地训练,但真的很慢。
希望你们能帮我摆脱这件事,我被困在塞维尔的日子里。
非常感谢!
发布于 2017-05-08 09:11:03
在迁移到新的theano后端时,我遇到了类似的问题,并通过以下步骤(在我的conda环境中)修复了它。
conda update theano更新theano,用conda install -c rdonnelly pygpu更新pygpy。我正在安装一个RC软件包,因为它提供了较少的问题。device = gpu更改为device = cuda0。如果您访问了上面的链接,您可能已经这样做了。https://stackoverflow.com/questions/43360279
复制相似问题