【发布时间】: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,conv2d和pool_2d都不能正常工作?结果我训练的时候错误率一直保持在90%左右。我是否正确安装了cudnn?我只是关注 theano 文档中的instructions,但 zip 中没有 .so* 文件。
我试过cpu模式,可以正常训练但是真的很慢。
希望你们能帮助我摆脱困境。
非常感谢!
【问题讨论】:
-
你的问题解决了吗?
标签: theano theano-cuda cudnn