Linux GPU 环境配置（驱动、Cuda和Cudnn安装）

注意，由于Nvidia 驱动、Cuda和Cudnn的版本实时更新，因此本文中涉及版本号的均由‘x’代替！！！

驱动安装

驱动下载

点击链接，根据显卡型号、操作系统下载相应驱动 Nvidia Driver Download Page

禁用 nouveau

打开终端，输入以下：

sudo gedit /etc/modprobe.d/blacklist.conf 

在 blacklist.conf 文件末尾添加并保存：

blacklist nouveau

然后执行命令更新：

sudo update-initramfs -u

重启并验证是否禁用了 nouveau：

lsmod | grep nouveau

无任何信息返回，则已被禁用。

驱动安装

卸载已安装的 Nvidia 驱动：

sudo apt-get remove --purge nvidia*

执行安装文件：

sudo bash ./NVIDIA-Linux-x86_64-xxx.xx.run -no-x-check -no-nouveau-check -no-opengl-files

验证是否安装成功：

nvidia-smi

Cuda 安装

Cuda下载

点击链接，根据系统类型，系统版本下载相应文件 Nvidia Cuda Download Page

依次点击：“Linux” -> “x86_64” -> “Ubuntu” -> “22.04” -> “runfile (local)”

这里作者建议采用"runfile (local)"的方式，简单快捷，不易出错。

此时，下载页面会显示如下执行信息：

wget https://developer.download.nvidia.com/compute/cuda/11.7.1/local_installers/cuda_11.7.1_515.65.01_linux.run
sudo sh cuda_11.7.1_515.65.01_linux.run

Cuda安装

在安装时，切记取消选择驱动安装！！！ 因上述步骤中，已经安装了最新版本的驱动。

执行完上述文件后，需将cuda中的bin目录添加到Path里，lib目录添加到LD_LIBRARY_PATH里。

打开环境变量配置文件：

vim ~/.bashrc

添加环境变量到文件底部：

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

保存并退出，验证Cuda是否安装成功：

nvcc -V

Cudnn安装

Cudnn下载

点击链接，根据Cuda版本下载相应文件 Nvidia Cudnn Download Page

这里，作者建议下载 “Local Installer for Linux x86_64 (Tar)”类型文件。

Cudnn安装

解压tar文件：

tar -xvf cudnn-linux-xxx-archive.tar

进入解压后的文件夹，并将lib下所有文件复制到cuda lib64文件夹下，include下所有文件复制到cuda include文件夹下：

cd cudnn
sudo cp lib/* /usr/local/cuda-11.x/lib64/
sudo cp include/* /usr/local/cuda-11.x/include/

查看并验证Cudnn版本信息：

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