[Python知识库]【Linux】Ubuntu 20.04 安装NVIDIA驱动 + CUDA + CUDNN

1. 安装NVIDIA驱动

首先检查系统推荐安装的版本

ubuntu-drivers devices

安装推荐版本驱动

sudo apt install nvidia-driver-XXX

安装完毕后重启，检查

nvidia-smi

2. 安装CUDA

从NVIDIA官网选择合适版本下载

推荐使用runfile安装

?在安装过程中选择Continue，输入accept，使用空格去掉Driver，最后install安装

安装完毕之后修改路径文件

gedit ~/.bashrc

在最后加上以下两句，注意修改CUDA版本

export PATH=/usr/local/cuda-1X.X/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-1X.X/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存路径

source ~/.bashrc

查看CUDA版本以验证安装成功

nvcc -V

3. 安装CUDNN

从官网中下载合适版本，需要注册登录NVIDIA账号

下载完之后，解压文件夹，并复制文件（注意修改路径）：

将cuda/include/cudnn.h文件复制到usr/local/cuda/include文件夹

sudo cp path/cuda/include/cudnn.h /usr/local/cuda/include

将cuda/lib64/下所有文件复制到/usr/local/cuda/lib64文件夹

sudo cp path/cuda/lib64/* /usr/local/cuda/lib64

复制完毕之后，添加读取权限

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

验证安装成功:

首先查看CUDNN版本文件路径

cat /usr/local/cuda/include/cudnn.h | grep cudnn
find / -name cudnn_version.h 2>&1 | grep -v "Permission denied"

使用以下命令查看CUDNN版本（注意修改文件路径）

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

感谢以下作者：

@ LUKE [链接]
@?ashome123 [链接]
@?哈希Ma [链接]
@ eaxy_z [链接]