一、安装显卡驱动

1.更新本地软件

sudo apt-get update

2.查看系统推荐的显卡驱动

ubuntu-drivers devices

 3.发现nvidia-driver-470是推荐的显卡驱动，所以输入以下命令

sudo apt-get install nvidia-driver-470

根据系统推荐灵活改变安装的版本

4.重启电脑，输入以下命令,如图所示则代表安装成功

nvidia-smi

 二、安装cuda10.2，其他版本同理

1.进入cuda下载网址,找到10.2版本并如图选择信息

2.进入终端，分别输入

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

出现如下界面，选择continue

 随后输入accept

最后一步，由于前面已经安装了驱动，要把安装驱动的选项取消掉，即没有叉号，如图所示，点击回车即可取消 ，最后选择install

 3.配置环境变量

 在终端中输入sudo gedit ~/.bashrc，将下列代码添加在文件底部，保存并退出。

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

最后更新一下环境变量，使软连接生效

sudo source ~/.bashrc

4.验证cuda是否安装成功

nvcc -V

若如下图所示，则代表安装成功

 三、安装cudnn8.0.5

1.进入cudnn官网，选择cudnn8.0.5，没有账号须注册才能下载

选择第一个进行下载

 下载后，文件出现在下载里，我在下载解压的时候出现无法提取到此处，把文件名改为

cudnn-10.2-linux-x64-v8.0.5.39.tgz

将改好名字的压缩包复制到主目录下，输入以下命令进行解压，会得到一个cuda的文件夹

tar -zxvf cudnn-10.2-linux-x64-v8.0.5.39.tgz

2.复制cuDNN内容到cuda相关文件夹内

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

3.检查是否安装成功，输入以下代码，结果如图所示，则代表安装成功

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