安装Nvidia-510驱动

目前用的是apt install的方式安装,使用中没有发现什么问题。

安装cuda10.2

1 下载

我这里用的ubuntu20.04,由于没有指定版本,下载的18.04的版本

cuda下载链接:
https://developer.nvidia.com/cuda-downloads
cuda历史版本下载链接:
https://developer.nvidia.com/cuda-toolkit-archive
(或点击 ‘Archive of Previous CUDA Releases’进入下载页面)

2 安装

有的博主说需要关闭图形界面,但我是直接在文件所在目录的终端里安装成功的

# Ctrl+Alt+F1 退出图形界面
sudo service lightdm stop

sudo sh ./cuda_10.2.89_440.33.01_linux.run
如果提示gcc版本问题:
Failed to verify gcc version.
See log at /var/log/cuda-installer.log for details.
解决方式1:cat /var/log/cuda-installer.log 则会看到问题及解决方法,执行时加上 --override 来忽略gcc版本不匹配问题。

sudo sh cuda_10.2.89_440.33.01_linux.run --override

3 配置cuda环境变量

gedit ~/.bashrc
# 在末尾加上以下内容:
export PATH=/usr/local/cuda-10.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH

4 保存退出后使环境变量生效:

source ~/.bashrc

查看cuda版本信息

nvcc -V

安装cuDNN8.5

1 下载

下载链接:
https://developer.nvidia.com/rdp/cudnn-download

下载 cuda10.2 对应的版本,解压,进入解压后的文件夹打开终端

2 解压

tar -xzvf {cudnn文件名}.tgz

3 复制文件

cd cudnn-linux-x86_64-8.5.0.96_cuda10-archive
sudo cp include/cudnn*.h /usr/local/cuda/include
sudo cp -P lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

4 查看cudnn版本信息

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