ubuntu18.04系统安装cuda、cudnn

1.安装Nvidia显卡驱动

1.1添加源并更新

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

1.2查询网卡驱动

ubuntu-drivers devices

1.3安装

sudo apt-get install nvidia-driver-510-server

1.4验证安装

nvidia-smi

2.安装cuda

2.1查看显卡驱动版本对应的cuda版本
官网链接：https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
我NVIDIA-SMI 510.47.03下的cuda10.2版本，根据自己需求选择版本
2.2下载并安装cuda
官网链接：https://developer.nvidia.com/cuda-toolkit-archive
打开终端执行：

wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run

#下载完后执行：sudo sh cuda_10.2.89_440.33.01_linux.run
会有界面弹出，如果你在之前安装好了显卡驱动，会让你移除并重新安装，因此我没有使用这个指令安装，我使用的是这个指令：通过加参数，在安装的过程中可能就不会检测驱动，然后成功安装

sudo bash cuda_10.2.89_440.33.01_linux.run --toolkit --silent --override

2.3添加环境变量
打开~/.bashrc，根据自己的路径添加：

# cuda
export PATH="/usr/local/cuda-10.2/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH"

让配置生效：source ~/.bashrc
2.4验证安装

nvcc -V

3.安装cudnn

官网链接：https://developer.nvidia.com/rdp/cudnn-archive
3.1下载
一定要选择跟自己cuda版本匹配的！我选择的是cudnn8.1.0，点击下载
3.2安装

#解压
tar -xzvf  cudnn-10.2-linux-x64-v8.1.0.77.tgz
#会在当前文件夹下生成一个名为cuda的文件夹，将里面的文件拷贝到cuda的安装目录下
sudo cp  cuda/lib64/libcudnn* /usr/local/cuda-10.2/lib64/
sudo cp  cuda/include/cudnn* /usr/local/cuda-10.2/include/
#修改权限
sudo chmod a+r /usr/local/cuda-10.2/include/cudnn.h 
sudo chmod a+r /usr/local/cuda-10.2/lib64/libcudnn*

3.3验证安装

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

参考连接：
https://blog.csdn.net/weixin_43563233/article/details/114385130
https://blog.csdn.net/wenroudebaozi/article/details/114582502
https://shliang.blog.csdn.net/article/details/108994449