Ubuntu20.04安装Nvidia驱动、Cuda10.2、Cudnn8.5
安装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