Ubuntu 24.04安装NVIDIA驱动与CUDA工具包完整指南

📅 2026/7/17 4:22:52 👁️ 阅读次数 📝 编程学习
Ubuntu 24.04安装NVIDIA驱动与CUDA工具包完整指南

1. Ubuntu 24.04下NVIDIA驱动与CUDA工具包安装指南

在Ubuntu 24.04上配置NVIDIA显卡驱动和CUDA工具包是进行深度学习、科学计算和图形处理的基础工作。作为一名长期使用Linux系统的开发者,我经历过无数次驱动安装失败、CUDA版本冲突等问题。本文将分享我在Ubuntu 24.04上安装NVIDIA驱动和CUDA工具包的完整流程,包括常见问题的解决方案和实用技巧。

2. 准备工作与环境检查

2.1 系统要求确认

在开始安装前,我们需要确认几个关键点:

  1. 确认你的系统确实配备了NVIDIA显卡。可以通过以下命令检查:
lspci | grep -i nvidia
  1. 检查当前系统版本是否为Ubuntu 24.04:
lsb_release -a
  1. 确保系统已更新到最新状态:
sudo apt update && sudo apt upgrade -y

2.2 禁用Nouveau驱动

Ubuntu默认使用开源的Nouveau驱动来支持NVIDIA显卡,我们需要先禁用它以避免冲突:

  1. 创建禁用配置文件:
sudo nano /etc/modprobe.d/blacklist-nouveau.conf
  1. 添加以下内容:
blacklist nouveau options nouveau modeset=0
  1. 更新initramfs并重启:
sudo update-initramfs -u sudo reboot

重启后,可以通过以下命令验证Nouveau是否已禁用:

lsmod | grep nouveau

如果没有输出,则表示禁用成功。

3. 安装NVIDIA显卡驱动

3.1 选择驱动安装方式

Ubuntu 24.04提供了三种安装NVIDIA驱动的方法:

  1. 使用Ubuntu官方仓库的驱动(推荐新手)
  2. 使用NVIDIA官方.run文件(适合需要特定版本)
  3. 使用PPA仓库(适合需要最新驱动)

我推荐使用第一种方法,因为它最简单且稳定。

3.2 通过Ubuntu仓库安装驱动

  1. 首先查看可用的驱动版本:
ubuntu-drivers devices
  1. 安装推荐的驱动(通常标记为"recommended"):
sudo apt install nvidia-driver-535
  1. 安装完成后重启系统:
sudo reboot

3.3 验证驱动安装

重启后,运行以下命令验证驱动是否正常工作:

nvidia-smi

如果看到类似如下的输出,说明驱动安装成功:

+-----------------------------------------------------------------------------+ | NVIDIA-SMI 535.86.05 Driver Version: 535.86.05 CUDA Version: 12.2 | |-------------------------------+----------------------+----------------------+ | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. | | | | MIG M. | |===============================+======================+======================| | 0 NVIDIA GeForce ... Off | 00000000:01:00.0 On | N/A | | 0% 43C P8 10W / 120W | 256MiB / 8192MiB | 0% Default | | | | N/A | +-------------------------------+----------------------+----------------------+

如果遇到"nvidia-smi has failed because it couldn't communicate with the nvidia driver"错误,请参考第6章的故障排除部分。

4. 安装CUDA工具包

4.1 选择CUDA版本

目前NVIDIA CUDA的最新稳定版本是12.8,但建议根据你的具体需求选择版本:

  • 深度学习框架(如TensorFlow、PyTorch)通常有特定的CUDA版本要求
  • 较新的CUDA版本可能不支持较旧的显卡

可以通过NVIDIA官网查看CUDA版本与显卡的兼容性。

4.2 通过官方仓库安装CUDA

  1. 添加NVIDIA CUDA仓库密钥:
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/3bf863cc.pub
  1. 添加仓库:
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/ /"
  1. 更新软件包列表:
sudo apt update
  1. 安装CUDA工具包(以12.8为例):
sudo apt install cuda-12-8

4.3 配置环境变量

安装完成后,需要将CUDA添加到系统路径中:

  1. 编辑~/.bashrc文件:
nano ~/.bashrc
  1. 在文件末尾添加:
export PATH=/usr/local/cuda-12.8/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-12.8/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
  1. 使更改生效:
source ~/.bashrc

4.4 验证CUDA安装

运行以下命令验证CUDA是否安装成功:

nvcc --version

如果看到类似如下的输出,说明CUDA安装成功:

nvcc: NVIDIA (R) Cuda compiler release 12.8, V12.8.89

5. 安装cuDNN(可选)

如果你计划进行深度学习开发,还需要安装cuDNN库:

  1. 从NVIDIA官网下载对应版本的cuDNN(需要注册账号)
  2. 解压并安装:
tar -xvf cudnn-linux-x86_64-8.x.x.x_cudaX.Y-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*

6. 常见问题与解决方案

6.1 "nvidia-smi has failed"错误

这是最常见的驱动问题,可能的原因和解决方案:

  1. 驱动未正确安装:
  • 重新安装驱动
  • 确保没有安装多个版本的驱动
  1. Secure Boot阻止驱动加载:
  • 进入BIOS禁用Secure Boot
  • 或为驱动签名
  1. 内核模块未加载:
sudo modprobe nvidia

6.2 安装后无法进入图形界面

如果安装驱动后无法进入桌面环境:

  1. 尝试在启动时选择恢复模式
  2. 卸载驱动:
sudo apt purge nvidia* sudo reboot
  1. 重新安装推荐版本的驱动

6.3 CUDA安装失败

常见原因包括:

  1. 系统依赖不满足:
sudo apt install build-essential
  1. 磁盘空间不足:
  • 清理/usr/local空间
  • 或指定其他安装目录
  1. 显卡不支持该CUDA版本:
  • 检查显卡计算能力
  • 选择更旧的CUDA版本

7. 性能优化与维护

7.1 持久模式设置

对于服务器环境,建议启用持久模式:

sudo nvidia-smi -pm 1

7.2 自动更新管理

为避免自动更新破坏驱动:

  1. 固定驱动版本:
sudo apt-mark hold nvidia-driver-535
  1. 禁用自动更新:
sudo apt-get remove unattended-upgrades

7.3 多GPU管理

对于多GPU系统,可以使用:

nvidia-smi topo -m

查看GPU拓扑结构,优化任务分配。

8. 实际使用建议

根据我的经验,以下几点值得注意:

  1. 在升级Ubuntu系统前,先卸载NVIDIA驱动,升级完成后再重新安装

  2. 保持驱动和CUDA版本的对应关系,避免版本冲突

  3. 对于生产环境,建议使用长期支持(LTS)版本的驱动和CUDA

  4. 定期检查NVIDIA官网的安全公告,及时更新驱动修复漏洞

  5. 对于笔记本用户,可能需要额外配置Optimus技术以在集成显卡和独立显卡间切换

安装完成后,你可以开始使用GPU加速的应用开发了。如果计划使用深度学习框架,建议先验证框架与CUDA版本的兼容性,再安装对应的版本。