在我使用线性回归做例子的时候GPU能正常运行,但在使用CNN时,神奇的事情发生了:
Loaded runtime CuDNN library: 5005 (compatibility version 5000) but source was compiled with 5105 (compatibility version 5100). If using a binary install, upgrade your CuDNN library to match. If building from sources, make sure the library loaded at runtime matches a compatible version specified during compile configuration.
查了博客后,最终解决了。主要问题就是CuDNN版本太低。我当时安装的是5.0(因为懒得自己申请所以网上找了一个。)后面还是自己去注册申请了,下载最新的版本终于得到解决。
https://developer.nvidia.com/rdp/cudnn-download
我的配置是1080 cuda - 8.0 所以下载了一个
Download cuDNN v5.1 (Jan 20, 2017), for CUDA 8.0
得到解决。
上个百度云,给需要的朋友
http://pan.baidu.com/s/1dFwwK29