ubuntu 16.04 安装CUDNN 6.0

一、前言:

安装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博客

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

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值