安装cudnn必须要看好版本下载,如果版本不匹配出现什么问题我暂时还不清楚,为了避免采坑还是按照搭配原则来吧,并且安装它之前要先安装好cuda
cuDNN Archive | NVIDIA Developer
因为我安装的CUDA是8.0版本的,CUDNN需要配合CUDA的版本,我选择的版本如下:
下载下来的文件后缀为.solitairetheme8 ,需要先将这个文件解压:
cp cudnn-8.0-linux-x64-v5.1.solitairetheme8 cudnn-8.0-linux-x64-v5.1.tgz
tar -zxvf cudnn-8.0-linux-x64-v5.1.tgz
cd cuda
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/* /usr/local/cuda/lib64/
cd /usr/local/cuda/lib64
如果安装其他版本的cudnn,在解压的时候会显示so文件的版本,替换下面的.so.5文件即可。
sudo rm -rf libcudnn.so libcudnn.so.5
sudo ln -s libcudnn.so.5.1.5 libcudnn.so.5
sudo vim /etc/profile
PATH=/usr/local/cuda/bin:$PATH
export PATH
sudo vim /etc/ld.so.conf.d/cuda.conf
在cuda.conf文件中添加以下信息并保存
/usr/local/cuda/lib64
sudo ldconfig
如果出现/sbin/ldconfig.real:/usr/local/cuda/lib64/libcudnn.so.7 不是符号连接,此时的解决方法为:
在终端输入:
sudo ldconfig -v
在输出的内容中找到这一行错误:libcudnn.so.7 -> libcudnn.so.7.0.5
错误的原因是这一行出现了问题,在终端输入这句命令:
sudo ln -sf /usr/local/cuda/lib64/libcudnn.so.7.0.5 /usr/local/cuda/lib64/libcudnn.so.7
nvidia-smi
或者查看版本号
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
感谢:Ubuntu16.04安装cuda+cudnn+Tensorflow踩坑全过程_ubuntu cuda cudnn tensorflow-CSDN博客
Ubuntu安装Tensorflow(GPU版)_ubuntu22安装tensorflow1-CSDN博客