Ubuntu20.04+CUDA11.1+cuDNN的详细配置过程

Ubuntu20.04+CUDA11.1+cuDNN的详细配置过程（NVIDIA RTX 3060Ti ）

安装CUDA11.1

下载安装包，可以从NVIDIA官网CUDA下载页面，链接为: https://developer.nvidia.com/cuda-toolkit-archive

我是参考一个博主的网盘链接，直接下载的链接:
网盘链接：https://link.csdn.net/?target=https%3A%2F%2Fpan.baidu.com%2Fs%2F1zIAHBIiQLxJBmQnKwDcv-Q
密码: bo6f

赋予权限执行

下载完成后，把安装包放在主目录下，打开终端，输入以下语句：

1.赋予安装包权限

sudo chmod +x cuda_11.1.0_455.23.05_linux.run

2.执行

sudo ./cuda_11.1.0_455.23.05_linux.run

设置环境变量

1.打开bashrc文件

sudo gedit ~/.bashrc

2.在bashrc文件最后加上下面语句，保存退出

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

3.执行bashrc文件

source ~/.bashrc

安装cuDNN

网盘链接: https://link.csdn.net/?target=https%3A%2F%2Fpan.baidu.com%2Fs%2F1rD72QvW2afMznW8mJ7GZpQ
密码: 935g

1.解压cudnn-11.1-linux-x64-v8.0.5.39.tgz文件

tar xvf cudnn-11.1-linux-x64-v8.0.5.39.tgz

2.赋予权限解压后的安装包权限

sudo cp cuda/include/cudnn.h /usr/local/cuda-11.1/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.1/lib64/
sudo chmod a+r /usr/local/cuda-11.1/include/cudnn.h 
sudo chmod a+r /usr/local/cuda-11.1/lib64/libcudnn*
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
sudo cp cuda/include/cudnn_version.h /usr/local/cuda-11.1/include/
sudo chmod a+r /usr/local/cuda-11.1/include/cudnn_version.h 
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

软链接

1.删除目前的软链接

sudo rm -rf /usr/local/cuda

2.生成新的软链接

sudo ln -s /usr/local/cuda-11.1 /usr/local/cuda

3.查看当前CUDA的版本

nvcc -V

或

nvcc --version

查看版本，第四行release 11.1就是安装的CUDA的版本

参考博客

参考博客链接: http://t.csdn.cn/QeJCb