Ubuntu 24.04深度学习环境配置指南

1. Ubuntu深度学习环境搭建全记录

作为一名长期在Linux环境下工作的算法工程师，我深知搭建深度学习环境的痛苦。每次换新机器或重装系统，都要重新折腾一遍驱动、CUDA和框架的安装。最近在Ubuntu 24.04上配置环境时，我详细记录了全过程，特别补充了那些教程里很少说明的细节问题。

这个配置方案适用于NVIDIA显卡的台式机和工作站，涵盖驱动安装、CUDA工具链配置、cuDNN部署以及PyTorch环境搭建。不同于官方文档的简略说明，我会重点解释每个步骤的技术原理和避坑要点，特别是那些容易导致安装失败的隐藏细节。

2. 系统基础环境准备

2.1 系统源配置与更新

刚安装完Ubuntu系统后，第一件事就是更换软件源。国内用户使用默认源下载速度很慢，建议切换为国内镜像源。我推荐清华或阿里云的源，不仅速度快而且同步及时。

在图形界面操作：

1. 打开"软件和更新"应用
2. 在"Ubuntu软件"选项卡中，找到"下载自"下拉菜单
3. 选择"mirrors.tuna.tsinghua.edu.cn"（清华源）或"mirrors.aliyun.com"（阿里云源）

注意：不同Ubuntu版本对应的源路径不同，24.04的代号是Noble Numbat，确保选择的源包含该版本支持

终端验证命令：

bash复制sudo apt update
sudo apt upgrade -y

这个步骤会更新所有已安装的软件包。建议在安装任何新软件前先执行更新，避免依赖冲突。

2.2 NVIDIA驱动安装

深度学习环境的核心是显卡驱动。Ubuntu自带的nouveau开源驱动性能有限，必须安装官方闭源驱动才能发挥GPU的全部能力。

首先检测显卡型号和推荐驱动：

bash复制ubuntu-drivers devices

输出示例：

code复制== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00002504sv000010DEsd00001517bc03sc00i00
vendor   : NVIDIA Corporation
model    : GA104 [GeForce RTX 3070 Ti]
driver   : nvidia-driver-535-server - distro non-free
driver   : nvidia-driver-535 - distro non-free recommended
driver   : nvidia-driver-550 - third-party non-free
driver   : nvidia-driver-470 - distro non-free
dri