一、前言:
安装cudnn必须要看好版本下载,如果版本不匹配出现什么问题我暂时还不清楚,为了避免采坑还是按照搭配原则来吧,并且安装它之前要先安装好cuda
二、官网下载cuda对应版本的cudnn,地址:
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
四、切换路径,进入刚解压出来的文件夹cuda中
cd cuda
五、复制文件
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/* /usr/local/cuda/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
八、编辑profile,添加环境变量
sudo vim /etc/profile
九、进入profile后,在最后一行添加如下信息并保存
PATH=/usr/local/cuda/bin:$PATH
export PATH
十、创建cuda.conf文件
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博客