Ubuntu系统下对NVIDIA显卡进行相关配置（NVIDIA驱动+CUDA+cuDNN）

Ubuntu系统下对NVIDIA显卡进行相关配置（NVIDIA驱动+CUDA+cuDNN）

一.总体内容

1.NVIDIA驱动， CUDA，cuDNN 联系与区别

2.MX150,2080ti，共同可用的配置

3.安装确定好的配置

---

二.NVIDIA驱动， CUDA，cuDNN 联系与区别

1.NVIDIA显卡只有成功安装驱动后才能使用。

2.CUDA 用于控制显卡进行并行计算。

3.cuDNN专门用于提高深度学习算法在显卡上的计算速度。它是基于CUDA框架开发的。

---

三.MX150,2080ti，共同可用的配置

通过查询不难知nvidia-driver-510 + CUDA11.6 + cuDNN v8.4.0的组合是可行的。

小贴士：Ubuntu（Linux）上如何查询显卡，输入以下指令。

lshw -c video

显然这是一块NVIDIA公司产的GeForce MX150显卡

---

四.安装确定好的配置

1.安装NVIDIA驱动

step1：禁用系统自带的nouveau显卡驱动

系统会自带驱动，因此安装NVIDIA驱动之前需要禁止系统自带显卡驱动nouveau。可先通过指令

lsmod | grep nouveau

查看nouveau驱动的启用情况，如果有输出则表示nouveau驱动正在工作，否则表示nouveau已经被禁用了。也可以通过 软件和更新->附加驱动 来查看。

禁用nouveau的操作为，输入以下指令

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

在打开的blacklist.conf文件末尾加上以下两行，并保存：

blacklist nouveau
options nouveau modeset=0

然后再输入以下指令，即可完成禁用系统自带的nouveau显卡驱动。

sudo update-initramfs -u 

step2：安装新的驱动

Ubuntu系统安装NVIDIA驱动相对方便，软件和更新->附加驱动->选择NVIDIA驱动（nvidia-driver-510）->应用更改，静候安装即可。

step3：验证NVIDIA驱动

安装完成后重启，输入以下指令，验证驱动安装结果。

nvidia-smi

在输出中可见Driver Version：510.85.02，表明安装成功。

与此同时可见CUDA Version：11.6 ，表明该版本的NVIDIA驱动支持的CUDA最高版本为11.6。

2.安装CUDA

step1:安装依赖库文件

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

step2：到NVIDIA官网CUDA下载页面找到目标CUDA版本，并安装

此处为CUDA 11.6,点击进入，进行进一步选择

Operating System：Linux

Architecture: x86_64

Distribution: Ubuntu

Version: 20.04

Installer Type: runfile[local]

完成选择后，依次执行官方提供的两条指令

wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux.run
sudo sh cuda_11.6.0_510.39.01_linux.run

在弹出的页面中，通过方向键选中Continue，回车，再输入accept。

然后再在新弹出的页面中，通过方向键选中Driver，回车（这一步为了避免覆盖掉前面安装的NVIDA驱动），然后再通过方向键选中Install，回车。静待安装完成。

step3：配置CUDA环境变量

配置环境变量后CUDA才能正常使用，输入以下命令

sudo gedit ~/.bashrc

在打开的.bashrc文件最后添加下面的CUDA环境变量配置信息（注意：不同版本的CUDA要把版本号改一下），保存并退出即可。

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

然后在终端输入下面指令或重新启终端使之生效。

source ~/.bashrc

step4：测试CUDA

输入以下命令即可检查CUDA信息，能看见所安装的版本号即为成功。

nvcc -V

3.安装cuDNN

step1：进入cuDNN官网下载链接（初次使用可能需要注册）

step2：找到对应CUDA版本的cuDNN（此处为cuDNN v8.4.0），点击打开，选择Local Installer for Linux x86_64[Tar]，等待下载完成。

step3：找到文件下载的位置（一般为 主目录->下载 目录） ，在该路径下打开终端，执行以下命令对所下文件进行解压并将.h 头文件都复制到 /usr/local/cuda 中，即完成安装。

tar -xvf cudnn-linux-x86_64-8.4.0.27_cuda11.6-archive.tar.xz
sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include 
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

step4：cuDNN测试

输入以下命令，若有cuDNN相关信息即为安装成功。

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