我的配置环境:win7+Anaconda3-4.4.0+python3.6+cuda9.0+cudnn7.1+tensorflow-gpu1.7
问题:之前一直用rnn或者dnn系列做一些自然语言以及推荐相关的工作,用显卡跑没什么问题。最近跑了个图像相关的cnn网络,立刻就报错:Loaded runtime CuDNN library: 7104 (compatibility version 7100) but source was compiled with 7003。这也说明了rnn和dnn没有用到cudnn加速。。。
解决方案:这是由于cudnn版本和tensorflow版本不匹配造成的,搜了一下解决方案主要有两种:(试了第一种有效,第二种没有测试)
(1)将cudnn7.1换成7.0.5;
(2)tensorflow-gpu1.7升级为1.8。