ubuntu 24.04 安装 Nvidia 显卡驱动 + CUDA + cuDNN，配置 AI 深度学习训练 ...

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

1.查看本机显卡型号

1. lspci | grep -i nvidia

复制代码

输出如下：
01:00.0 3D controller: NVIDIA Corporation GM108M [GeForce 920MX] (rev a2)
此中，GeForce 920MX就是我们的显卡型号。
2.下载 Nvidia 显卡驱动

官网：点击此处下载NVIDIA驱动
根据显卡型号搜索驱动：

选择查找

下载最新版驱动

3.安装显卡驱动

安装编译环境

1. sudo apt update
2. sudo apt install gcc make

复制代码

运行安装步伐

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

复制代码

选择“Continue installation”，回车

进入编译中…

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

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

复制代码

重新运行安装步伐：

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

复制代码

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

安装过程中，会提示是否禁用 nouveau 驱动，选择是，NVIDIA 会自动屏蔽 nouveau 驱动，不消手动禁止。
在安装过程中，没有不是特别要求的话，提示选择是否的话，可以都选择“是”。

终端运行nvidia-smi
：
输出如下，可以查看到版本号和显存：

安装完成，重启。
4.CUDA 安装

查看显卡支持的 CUDA 版本

1. nvidia-smi

复制代码

输出如下：

CUDA 最高可用版本为 12.6
到官网下载 CUDA：https://developer.nvidia.com/cuda-toolkit-archive
留意：CUDA 需要注册登录 NVIDIA 官网才气下载。

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

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

这里是：

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

复制代码

这里以 CUDA Toolkit 12.2.2为例安装:

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

复制代码

选择“Continue”

输入“accept”

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

等待安装完成。
配置环境

1. nano ~/.bashrc

复制代码

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

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

复制代码

重载配置

1. source ~/.bashrc

复制代码

检查安装

1. nvcc -V

复制代码

输出如下，则安装成功：

5.安装 cuDNN

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

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

1. 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
3. sudo dpkg -i cudnn-local-repo-ubuntu2404-9.3.0_1.0-1_amd64.deb
5. sudo cp /var/cudnn-local-repo-ubuntu2404-9.3.0/cudnn-*-keyring.gpg /usr/share/keyrings/
7. sudo apt-get update
9. 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包安装即可。

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。