一、安装显卡驱动
确认需要的 cuda 版本，以 cuda10.1 为例
1. 查看版本对应
在cuda说明文档中查看需要的cuda对应的显卡驱动版本
cuda10.1 在 linux 下需要显卡驱动版本 >=418.39
2. 在线安装显卡驱动
- 添加ppa源
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
- 查看添加的源中支持当前显卡的驱动版本
ubuntu-drivers devices
- 安装需要的显卡驱动，以 nvidia-430 为例，
sudo apt install nvidia-430
注意驱动名字，要与上图中需要的版本一致
- 查看显卡驱动信息
nvidia-smi
二、安装cuda
1.下载
去cudal工具包存档下载需要的 cuda 版本
- 选择第一种 runfile(local) 方式(其他方法也可以)
- 可以通过终端输入 wget 命令下载安装包
wget https://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
- 也可以复制上图选中的链接，手动下载安装包
2.安装
- 运行下载打 .run 文件进行安装
sudo sh cuda_10.1.243_418.87.00_linux.run
选择 continue 继续
下边输入 accept 继续
去掉 Driver 前面的 X，不安装 Driver，因为之前以及安装好了，选择 install 开始安装
3.添加环境变量
- 打开 .banshrc 文件
sudo gedit ~/.bashrc
- 在打开的文档最后添加下边三行，最后一行修改为自己的 cuda 版本
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
- 保存文件后退出，终端输入
source ~/.bashrc
4.测试是否安装成功
- 查看cuda版本两种方法，注意空格和大写 V
nvcc -V
cat /usr/local/cuda/version.txt
显示成功安装 cuda10.1.243
- 也可以用下面方法测试
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
三、安装cudnn
1.通过 cuDNN Library for Linux 安装
- 在cudnn官网下载所需的 cudnn，需注册登录
选择 cuDNN Library for Linux (x86) - 解压下载好的文件，下载好的文件名
tar -xzvf cudnn-10.1-linux-x64-v8.0.5.39.tgz
- 复制文件
sudo cp -r cuda/include/ /usr/local/cuda/include/
sudo cp -r cuda/lib64/ /usr/local/cuda/lib64/
- 赋予权限
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
- 查看 cudnn 版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
或者
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
根据 cudnn 版本不同，使用的命令不同，参考链接
