ubuntu 24.04 安装 Nvidia 显卡驱动 + CUDA + cuDNN，配置 AI 深度学习训练环境，简单易懂，一看就会！

ubuntu 24.04 安装 Nvidia 显卡驱动 + CUDA + cuDNN，配置 AI 深度学习训练环境，简单易懂，一看就会！

1.查看本机显卡型号

lspci | grep -i nvidia

输出如下：

01:00.0 3D controller: NVIDIA Corporation GM108M [GeForce 920MX] (rev a2)

其中，GeForce 920MX就是我们的显卡型号。

2.下载 Nvidia 显卡驱动

官网：点击此处下载NVIDIA驱动

根据显卡型号搜索驱动：

选择查找

下载最新版驱动

3.安装显卡驱动

安装编译环境

sudo apt update
sudo apt install gcc make

运行安装程序

chmod +x NVIDIA-Linux-x86_64-560.31.02.run
sudo ./NVIDIA-Linux-x86_64-560.31.02.run

选择“Continue installation”，回车

进入编译中…

注意：在这里编译时会出现编译错误而中断，原因是下载的显卡驱动是最新的，需要用较高的 gcc 版本编译器来编译，默认 gcc 的版本是 11，小于编译所需要的 12 版本。因此需要安装 12 版本的，并调整 gcc 链接。

sudo apt install gcc-12
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/x86_64-linux-gnu-gcc-12 20

重新运行安装程序：

sudo ./NVIDIA-Linux-x86_64-560.31.02.run

编译通过后，接着运行到下面这里，选择“Yes”

安装过程中，会提示是否禁用 nouveau 驱动，选择是，NVIDIA 会自动屏蔽 nouveau 驱动，不用手动禁止。

在安装过程中，没有不是特别要求的话，提示选择是否的话，可以都选择“是”。

终端运行nvidia-smi：

输出如下，可以查看到版本号和显存：

安装完成，重启。

4.CUDA 安装

查看显卡支持的 CUDA 版本

nvidia-smi

输出如下：

CUDA 最高可用版本为 12.6

到官网下载 CUDA：https://developer.nvidia.com/cuda-toolkit-archive

注意：CUDA 需要注册登录 NVIDIA 官网才能下载。

选择 12.6 版本以下的都可，最高版本可选择 12.6，根据系统版本选择相应的项：

得到下载地址和运行命令：

这里是：

wget https://developer.download.nvidia.com/compute/cuda/12.2.2/local_installers/cuda_12.2.2_535.104.05_linux.run

sudo sh cuda_12.2.2_535.104.05_linux.run

这里以 CUDA Toolkit 12.2.2为例安装:

chmod +x cuda_12.2.2_535.104.05_linux.run 
sudo ./cuda_12.2.2_535.104.05_linux.run 

选择“Continue”

输入“accept”

取消“Driver”选项，因为已经安装过显卡驱动了， 这里不需要安装，然后选择“Install”。

等待安装完成。

配置环境

nano ~/.bashrc

在文件最后添加以下内容：

export PATH=$PATH:/usr/local/cuda-12.2/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-12.2/lib64 

重载配置

source ~/.bashrc

检查安装

nvcc -V

输出如下，则安装成功：

5.安装 cuDNN

官网下载：https://developer.nvidia.com/rdp/cudnn-download

选择相应项后会生成下载命令：

wget https://developer.download.nvidia.com/compute/cudnn/9.3.0/local_installers/cudnn-local-repo-ubuntu2404-9.3.0_1.0-1_amd64.deb

sudo dpkg -i cudnn-local-repo-ubuntu2404-9.3.0_1.0-1_amd64.deb

sudo cp /var/cudnn-local-repo-ubuntu2404-9.3.0/cudnn-*-keyring.gpg /usr/share/keyrings/

sudo apt-get update

sudo apt-get -y install cudnn

一般最新版本的显卡驱动都能适配最新版本的 cuDNN，若最新版本不适合当前的 CUDA，可以安装历史版本。

cuDNN 最新版本支持的 CUDA 查看：Support Matrix — NVIDIA cuDNN v9.3.0 documentation

查找 cuDNN 支持的 CUDA 的历史版本:

选择"cuDNN 8.x-1.x"

然后选择适应的版本下载deb包安装即可。