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