一、准备工作
Nvidia驱动的安装，看我另外一个文章ubuntu18.04 安装Nvidia驱动的三种方式（必看）_低吟浅笑-CSDN博客
打开终端，输入
nvidia-smi出来下列信息：
出来该信息，说明驱动无误。查看驱动显示出来的CUDA版本，所安装的版本尽量与这个接近，驱动可以向下兼容，也就是可以安装比显示出来的CUDA版本低的版本。如图所示
二、安装CUDA
1. 官网下载相应的版本，需要考虑nvidia驱动：CUDA Toolkit Archive | NVIDIA Developer
2. 根据 nvidia驱动的版本下载cuda，注意下面有个 wget 网址的选项，如果wget下载较慢，可以打开链接直接下载。
3. 安装 cuda
进入下载好的文件夹，输入
sudo sh cuda_10.0.130_410.48_linux.run注意，新版的cuda安装是个小窗口，需要按 Enter 选择哪个不安装，旧版是通过提问的方式来的。
第一个选项选 accept
除了安装 驱动 的选项，其他都选 yes
4.配置环境变量
输入下面的命令打开bashrc
sudo vi ~/.bashrc
或者
sudo gedit ~/.bashrc在最下面添加下面内容，注意 里面的 cuda-10.0 需要换成自己的版本，自己可以去 /usr/local/ 下面看一下呢
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64
export PATH=$PATH:/usr/local/cuda-10.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.0
重新启动配置文件
source ~/.bashrc三、安装CUDNN
登陆 cuDNN Archive | NVIDIA Developer，需要提前注册账号
根据cuda的版本找到对应的cudnn版本，linux下载tar包
不知道cuda版本用 nvcc -V查询，查不到就说明cuda的环境变量可能有问题，回退上一步
解压文件
// 注意自己下载文件的路径和名称
tar -xvf cudnn-linux-x86_64-*.tar.xz复制cudnn的文件，注意cudnn解压后的名字
# 进入cudnn的路径
cd cudnn-linux-x86_64-8.8.0.121_cuda11-archive/
# 查看自己的cuda版本，我的是11.4
ls /usr/local
# 复制文件
sudo cp include/cudnn.h /usr/local/cuda-11.4/include
sudo cp lib/libcudnn* /usr/local/cuda-11.4/lib64
sudo chmod a+r /usr/local/cuda-11.4/include/cudnn.h
sudo chmod a+r /usr/local/cuda-11.4/lib64/libcudnn*
注意一下:cudnn8.0以上会将版本信息更新到cudnn_version.h文件中,也需要将这个文件复制,否则验证的时候会没有反应
sudo cp include/cudnn_version.h /usr/local/cuda-11.4/include/
验证成功
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
出现上面的信息即为成功