cuDNN安装

下面以cuDNN8.2.0为例,基于ubuntu20.04安装,默认已经安装GPU驱动、CUDA。

1. 下载cuDNN

cuDNN8.2.0下载链接:https://developer.nvidia.com/rdp/cudnn-archive

根据已经安装的CUDA版本,选择相应的下载链接,因为已经安装了cuda11.3,因此选择了如下红框标注版本:

 2. 安装cuDNN

将下载好的软件包,上传至Linux操作系统任意用户目录下,例如将cuDNN包

cudnn-11.3-linux-x64-v8.2.0.53.tgz上传至~/software目录,并进入到该目录下:

1) 解压

tar xvf cudnn-11.3-linux-x64-v8.2.0.53.tgz

2)拷贝头文件、库文件

进入解压后的cuda目录下的,拷贝头文件、库文件到已安装的CUDA目录下:

cd cuda 
sudo cp */*.h /usr/local/cuda/include/ 
sudo cp lib64/libcudnn* /usr/local/cuda/lib64/

3)创建软连接

如果在执行步骤4)的时候出现软连接异常,需要重新创建软连接,进入到CUDA安装目录下,更新软连接:

cd /usr/local/cuda/lib64

sudo rm libcudnn.so libcudnn.so.8
sudo ln -s libcudnn.so.8.2.0 libcudnn.so.8
sudo ln -s libcudnn.so.8 libcudnn.so

sudo rm libcudnn_cnn_infer.so libcudnn_cnn_infer.so.8
sudo ln -s libcudnn_cnn_infer.so.8.2.0 libcudnn_cnn_infer.so.8
sudo ln -s libcudnn_cnn_infer.so.8 libcudnn_cnn_infer.so

sudo rm libcudnn_adv_train.so libcudnn_adv_train.so.8
sudo ln -s libcudnn_adv_train.so.8.2.0 libcudnn_adv_train.so.8
sudo ln -s libcudnn_adv_train.so.8 libcudnn_adv_train.so

sudo rm libcudnn_ops_infer.so libcudnn_ops_infer.so.8
sudo ln -s libcudnn_ops_infer.so.8.2.0 libcudnn_ops_infer.so.8
sudo ln -s libcudnn_ops_infer.so.8 libcudnn_ops_infer.so

sudo rm libcudnn_cnn_train.so libcudnn_cnn_train.so.8
sudo ln -s libcudnn_cnn_train.so.8.2.0 libcudnn_cnn_train.so.8
sudo ln -s libcudnn_cnn_train.so.8 libcudnn_cnn_train.so

sudo rm libcudnn_adv_infer.so libcudnn_adv_infer.so.8
sudo ln -s libcudnn_adv_infer.so.8.2.0 libcudnn_adv_infer.so.8
sudo ln -s libcudnn_adv_infer.so.8 libcudnn_adv_infer.so

sudo rm libcudnn_ops_train.so libcudnn_ops_train.so.8
sudo ln -s libcudnn_ops_train.so.8.2.0 libcudnn_ops_train.so.8
sudo ln -s libcudnn_ops_train.so.8 libcudnn_ops_train.so

4)使动态链接库生效

sudo ldconfig

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

洪流之源

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值