由于 cuda tensorflow-gpu cudnn 等版本号对应问题,在运行tensorflow-gpu时会出现错误,此时应到官网下载对应的cudnn版本,本文简要介绍更换cudnn版本号的方法:
第一步:官网下载cudnn(如cuda 9.0 对应 cudnn 7.6.2),解压出cuda文件夹
第二步:删除已有cudnn:
$ sudo rm -rf /usr/local/cuda/include/cudnn.h
$ sudo rm -rf /usr/local/cuda/lib64/libcudnn*
第三步:cd到解压的cuda文件夹下复制对应的文件到路径:
$ sudo cp include/cudnn.h /usr/local/cuda/include/
$ sudo cp lib64/lib* /usr/local/cuda/lib64/
第四步:添加文件路径,注意根据自己的版本号添加,本文以cudnn 7.6.2为例
$ sudo chmod +r libcudnn.so.7.6.2
$ sudo ln -sf libcudnn.so.7.6.2 libcudnn.so.7
$ sudo ln -sf libcudnn.so.7 libcudnn.so
$ sudo ldconfig
更换完成!再次运行代码,报错消失