我正在尝试配置theano
在我的Windows机器上使用gpu
。我已经设置.theanorc
使用device= gpu
但是当我运行一些应该使用gpu的代码时,我收到以下错误:
Can not use cuDNN on context None: cannot compile with cuDNN. We got this error:
c:\users\...\appdata\local\temp\try_flags_pt24sj.c:4:19: fatal error: cudnn.h: No such file or directory
compilation terminated.
Mapped name None to device cuda0: GeForce 840M (0000:03:00.0)
我检查了我的CUDA_PATH=C:\Program Files\NVIDIA\v8.0
GPU计算工具包\ CUDA,看看cudnn.h是否存在,我在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include
找到了它
修复此问题,遵循安装指南https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#install-windows并将LIBRARY_PATH添加到PATH变量(指向您的CUDA lib文件夹,例如C:\ Program Files \ NVIDIA GPU Computing Toolkit \ CUDA \ v9.0 \ lib)。