cuDNN安装

Posted 2023-02-26 洪流之源

下面以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