cuda 的官方网址和cudnn 的官方网址，cudnn 的下载需要注册账号。

安装 cuda

1）验证 cuda 的安装条件

首先，在终端输入lspci | grep -i _nvidia_ 查看你的 gpu 版本信息
接着，验证是否安装 gcc，终端输入 gcc –version ，如果未安装，则输入 sudo apt-get install build-essential 进行安装
最后，验证系统是否安装了 kernel header 和 package development
终端输入 uname -r 查看内核 接着输入 sudo apt-get install linux-headers-$(uname -r)
即 sudo apt-get install linux-headers-4.18.20-25-generic
就可以安装对应 kernel 版本的 kernel header 和 package development

2）安装 cuda

终端输入 sudo sh 你下载的 cuda.run 然后一直按住回车键，直到服务条款显示到 100%。除了 driver 选项选择 no 其他都选择 yes。

这里我直接用官网的方法安装的https://developer.nvidia.com/cuda-downloads

3）设置环境变量
在终端输入 sudo gedit ~/.bashrc
在文本的最后输入

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

目录要对应自己的 cuda 版本

4）然后 source ~/.bashrc 激活环境
5）在终端输入 nvcc -V 出现信息则安装成功

安装 cudnn

下载压缩包就可以了。我把个deb都下载安装后好像没什么作用。
官方教程： https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#download
官方验证方法：https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#verify

  1. Navigate to yourdirectory containing the cuDNN tar file.
  2. Unzip thecuDNNpackage.

$ tar -xzvf cudnn-x.x-linux-x64-v8.x.x.x.tgz
or
$ tar -xzvf cudnn-x.x-linux-aarch64sbsa-v8.x.x.x.tgz

  1. Copy the following files into theCUDA Toolkitdirectory.

$ sudo cp cuda/include/cudnn*.h /usr/local/cuda/include <br />$ sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64 <br />$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

To verify that cuDNN is installed and is running properly, compile the mnistCUDNN sample located in the/usr/src/cudnn_samples_v8 directory in the Debian file.

  1. Copy the cuDNN samples to a writable path.

$cp -r /usr/src/cudnn_samples_v8/ $HOME

  1. Go to the writable path.

$ cd $HOME/cudnn_samples_v8/mnistCUDNN

  1. Compile themnistCUDNNsample.

$make clean && make

  1. Run themnistCUDNNsample.

$ ./mnistCUDNN
If cuDNN is properly installed and running on your Linux system, you will see a message similar to the following:
Test passed!

参考博客：https://www.jianshu.com/p/158ae8fcdb4a