运行vgg16做transfer learning的例子,demo来源:https://machinelearningmastery.com/how-to-use-transfer-learning-when-developing-convolutional-neural-network-models/ 扯远了,回来
出现以下错误:
Loaded runtime CuDNN library: 7.3.0 but source was compiled with: 7.4.1.
看了一下自己的cuda以及cudnn版本,如下:
- 查看cuda版本
在以下目录:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\version.txt
我的版本是 CUDA Version 10.0.130 - 查看cudnn版本
找到以下目录:
1)C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include
2)找到该目录下cudnn.h文件,用记事本打开
3)ctrl+H搜索CUDNN_MAJOR
4)得知我的版本是cudnn 7.3.0
而我的tensorflow版本是1.13.0,搜了一圈,发现下图,详述了tensorflow,cuda, cudnn几个之间的匹配关系,如图。
原址在这里:https://www.tensorflow.org/install/source#common_installation_problems 非windows系统
https://www.tensorflow.org/install/source_windows 对于windows系统
去英伟达官网下载cudnn 7.4:
https://developer.nvidia.com/rdp/cudnn-archive
提示:下载cudnn需要注册NVIDIA账户,请自行注册并登陆
找到对应的版本,我的是: cuDNN v7.4.1 (Nov 8.2018). for CUDA 10.0,然后下载windows下的,文件大小为325M
下载解压完成后有四个文件夹:
将bin include lib里面的文件分别拷贝至CUDA安装环境下相应的文件夹,注意是里面的文件而不是整个文件夹。CUDA安装环境:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
bin里面的文件也就是cudnn64_7.dll拷贝至
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
的文件夹下,替换原有的.dll文件
完成替换以后,配置以下四个windows系统环境变量:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include
如果以前在安装CUDA时已经配置过,则这次可以省略此步。
完成上述操作后,问题解决!