ubuntu 16.04 安装CUDNN 6.0

一、前言:

安装cudnn必须要看好版本下载,如果版本不匹配出现什么问题我暂时还不清楚,为了避免采坑还是按照搭配原则来吧,并且安装它之前要先安装好cuda

ubuntu 16.04 安装CUDNN 6.0_第1张图片

二、官网下载cuda对应版本的cudnn,地址:

cuDNN Archive | NVIDIA Developer

因为我安装的CUDA是8.0版本的,CUDNN需要配合CUDA的版本,我选择的版本如下:

ubuntu 16.04 安装CUDNN 6.0_第2张图片

 三、下载之后,进入到文件目录中,解压文件包

下载下来的文件后缀为.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博客

Ubuntu安装Tensorflow(GPU版)_ubuntu22安装tensorflow1-CSDN博客

你可能感兴趣的:(环境配置,ubuntu,linux,运维)