我的tensorflow之前的版本是1.2的所以支持cudnn5,但是tensorflow1.3及以上就是支持cudnn6。
查看:
/usr/local/cuda/lib64$ ls libcudnn*
libcudnn.so libcudnn.so.5 libcudnn.so.5.1.5 libcudnn_static.a
因此要到官网下载cudnn6啦,官网链接:https://developer.nvidia.com/rdp/cudnn-download
我下载的是:cudnn-8.0-linux-x64-v6.0.tgz :https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v6/prod/8.0_20170307/cudnn-8.0-linux-x64-v6.0-tgz
然后解压到本地,发现出现cuda文件夹,然后:
1、覆盖之前cudnn5的头文件
$ cd cuda/include
$ sudo cp *.h /usr/local/cuda/include/ (注意这里是你自己的cuda的安装地址,复制*.h文件)
2、覆盖lib库,记得把cudnn5删掉
$ cd cuda/lib64
$ sudo rm /usr/local/cuda/lib64/libcudnn*
$ sudo cp libcudnn* /usr/local/cuda/lib64/
3(可选,好像不用下面这步骤也可以)
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*