cuDNN安装

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

1. 下载cuDNN

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

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

cuDNN安装_第1张图片

 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

你可能感兴趣的:(人工智能,python,深度学习)