tensorflow 升级后报错:ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory

时间:2021-07-11 11:45:34

我的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*