是的,Ubuntu 24.04 LTS(代号 Noble Numbat) 已经发布,并且非常适合用于搭建深度学习环境。它是一个长期支持版本(LTS),官方支持将持续到 2029 年,稳定性好、兼容性强,适合科研和开发使用。
✅ Ubuntu 24.04 搭建深度学习环境的优势:
- 内核更新(Linux 6.8):更好的硬件支持。
- Python 版本默认为 3.12:现代 Python 开发环境支持。
- 软件源丰富:支持主流深度学习框架如 PyTorch、TensorFlow、Jupyter、CUDA 工具链等。
- 良好的 GPU 支持:NVIDIA 显卡驱动、CUDA 和 cuDNN 都有官方支持。
- 适合容器化部署:Docker、Singularity 等工具支持良好,便于构建隔离环境。
🧰 推荐的深度学习环境组件
| 组件 | 推荐版本(Ubuntu 24.04 上) |
|---|---|
| OS | Ubuntu 24.04 LTS |
| Python | 3.10 / 3.11 / 3.12 |
| Conda(可选) | Miniconda / Anaconda |
| CUDA Toolkit | 12.4(对应 NVIDIA 驱动 >= 550) |
| cuDNN | v8.9.x |
| PyTorch | ≥ 2.3(支持 CUDA 12.4) |
| TensorFlow | ≥ 2.15(支持 CUDA 12.2+) |
| Docker | 官方推荐安装最新版 |
| NVIDIA 驱动 | 最新版(>=550)或根据显卡型号选择 |
🛠️ 搭建步骤概览
1. 安装系统
- 下载 Ubuntu 24.04 ISO
- 制作 U 盘启动盘,安装系统(建议选择英文语言环境避免部分编码问题)
2. 更新系统并安装基础工具
sudo apt update && sudo apt upgrade -y
sudo apt install build-essential cmake git curl wget unzip vim tmux -y
3. 安装 NVIDIA 显卡驱动
方法一:通过官方 PPA(推荐)
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
sudo apt install nvidia-driver-550 # 根据你的显卡型号选择
重启后执行:
nvidia-smi
查看是否成功识别 GPU。
4. 安装 CUDA Toolkit(以 12.4 为例)
你可以从官网下载 .deb 包安装,或者直接用 apt 安装(Ubuntu 24.04 的仓库已经包含较新版本):
sudo apt install cuda-toolkit-12-4
添加环境变量(写入 ~/.bashrc 或 ~/.zshrc):
export PATH=/usr/local/cuda-12.4/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.4/lib64:$LD_LIBRARY_PATH
验证:
nvcc --version
5. 安装 cuDNN(通常已随 CUDA 安装)
如果需要单独安装 cuDNN:
sudo apt install libcudnn8=8.9.7.29-1+cuda12.4
6. 安装 Python 和虚拟环境
使用系统 Python(推荐搭配 venv):
sudo apt install python3 python3-pip python3-venv -y
或者安装 Miniconda(更推荐):
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
chmod +x Miniconda3-latest-Linux-x86_64.sh
./Miniconda3-latest-Linux-x86_64.sh
创建虚拟环境示例:
conda create -n dl_env python=3.11
conda activate dl_env
7. 安装深度学习框架
安装 PyTorch(GPU 支持):
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
安装 TensorFlow(GPU 支持):
pip install tensorflow --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v1026/
注意:TensorFlow 在 Linux 上对 CUDA 支持不如 PyTorch 好,有时需要手动配置
LD_LIBRARY_PATH或使用 Docker。
8. 可选:安装 Jupyter Notebook / VSCode / 其他 IDE
pip install jupyter notebook
jupyter notebook
🧪 测试 GPU 是否可用
在 Python 中测试 PyTorch 是否能调用 GPU:
import torch
print(torch.cuda.is_available())
print(torch.__version__)
输出类似:
True
2.4.0+cu124
表示 GPU 正常工作!
💡 小贴士
-
如果你使用的是笔记本电脑或双显卡设备,请确保使用 NVIDIA GPU 而不是集显(可以通过
prime-select设置):sudo prime-select nvidia reboot -
遇到问题时可以查看日志文件
/var/log/Xorg.0.log或运行nvidia-bug-report.sh提交 bug。
✅ 总结
| 项目 | Ubuntu 24.04 支持情况 |
|---|---|
| CUDA | ✅ 支持(12.4) |
| cuDNN | ✅ 支持(v8.9) |
| PyTorch | ✅ 支持(2.4+) |
| TensorFlow | ✅ 支持(需注意依赖) |
| Conda | ✅ 支持 |
| Docker | ✅ 支持 |
如果你需要我帮你生成一个完整的脚本或配置文件来自动化安装,也可以告诉我你的具体需求(比如是否使用 conda?是否需要 TF?是否多用户?)我可以为你定制一个一键安装脚本。
是否需要我提供一个自动安装脚本模板?
秒懂云