Ubuntu18.04安装CUDA11.1+cudnn8.0.5

1、查看驱动

nvidia-smi

2. 安装cuda

wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux.run

sudo sh cuda_11.1.0_455.23.05_linux.run

选择，然后输入accept,进入安装选项 installer；X代表选择，按enter键取消X消失（即不选择）。
我因为装过驱动了，所以驱动那个选项不选，其他全选了，然后就是选install，进行安装，一会就安装好了。

配置环境变量

gedit ~/.bashrc

export PATH=/usr/local/cuda-11.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

source ~/.bashrc
验证是否安装成功

nvcc -V
3、安装cudnn

安装cudnn (cudnn官方下载地址)

sudo dpkg -i libcudnn8_8.0.5.39-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-dev_8.0.5.39-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-samples_8.0.5.39-1+cuda11.0_amd64.deb

测试是否安装成功，分别输入以下四个命令：
cp -r /usr/src/cudnn_samples_v8/ ~ 
cd ~/cudnn_samples_v8/mnistCUDNN/ 
make clean && make 
./mnistCUDNN
4.查看cuda、cudnn

nvcc -V

cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

完成安装