安装Nvidia CUDA ToolKit及cudnn

安装CUDA ToolKit

CUDA下载地址：

https://developer.nvidia.com/cuda-toolkit-archive

在页面中获取所需的cuda版本，然后执行安装命令(以11.8为例)

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sudo sh cuda_11.8.0_520.61.05_linux.run

安装完成后需要设置环境改变量，安装完成后也会有提示如何设置

export PATH=/usr/local/cuda-11.4/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH

cuda-11.4替换为具体的cuda版本

安装CuDNN

CuDNN下载地址：
https://developer.nvidia.com/rdp/cudnn-download

先从下载页面获取到CuDNN下载地址及文件名，然后使用wget下载CuDNN

wget http://developer.download.nvidia.com/compute/redist/cudnn/v8.9.2/cudnn-linux-x86_64-8.9.2.26_cuda11-archive.tar.xz

链接中v8.9.2替换成版本号，最后一段文件名部分替换为实际的文件名

CuDNN安装

sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h 
sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*

命令中的cuda-10.1替换为cuda版本

如果报libcudnn_cnn_train.so.8 is not a symbolic link的话执行以下命令

sudo ln -sf /usr/local/cuda-11.2/targets/x86_64-linux/lib/libcudnn_cnn_train.so.8.2.0 /usr/local/cuda-11.2/targets/x86_64-linux/lib/libcudnn_cnn_train.so.8
sudo ln -sf /usr/local/cuda-11.2/targets/x86_64-linux/lib/libcudnn_adv_infer.so.8.2.0 /usr/local/cuda-11.2/targets/x86_64-linux/lib/libcudnn_adv_infer.so.8
sudo ln -sf /usr/local/cuda-11.2/targets/x86_64-linux/lib/libcudnn_adv_train.so.8.2.0 /usr/local/cuda-11.2/targets/x86_64-linux/lib/libcudnn_adv_train.so.8
sudo ln -sf /usr/local/cuda-11.2/targets/x86_64-linux/lib/libcudnn_cnn_infer.so.8.2.0 /usr/local/cuda-11.2/targets/x86_64-linux/lib/libcudnn_cnn_infer.so.8
sudo ln -sf /usr/local/cuda-11.2/targets/x86_64-linux/lib/libcudnn.so.8.2.0 /usr/local/cuda-11.2/targets/x86_64-linux/lib/libcudnn.so.8
sudo ln -sf /usr/local/cuda-11.2/targets/x86_64-linux/lib/libcudnn_ops_infer.so.8.2.0 /usr/local/cuda-11.2/targets/x86_64-linux/lib/libcudnn_ops_infer.so.8
sudo ln -sf /usr/local/cuda-11.2/targets/x86_64-linux/lib/libcudnn_ops_train.so.8.2.0 /usr/local/cuda-11.2/targets/x86_64-linux/lib/libcudnn_ops_train.so.8

命令中的 cuda-11.2替换为实际版本，8.2.0也替换为实际版本