ubuntu 24.04深度学习环境搭建可以吗?

是的,Ubuntu 24.04 LTS(代号 Noble Numbat) 已经发布,并且非常适合用于搭建深度学习环境。它是一个长期支持版本(LTS),官方支持将持续到 2029 年,稳定性好、兼容性强,适合科研和开发使用。


✅ Ubuntu 24.04 搭建深度学习环境的优势:

  1. 内核更新(Linux 6.8):更好的硬件支持。
  2. Python 版本默认为 3.12:现代 Python 开发环境支持。
  3. 软件源丰富:支持主流深度学习框架如 PyTorch、TensorFlow、Jupyter、CUDA 工具链等。
  4. 良好的 GPU 支持:NVIDIA 显卡驱动、CUDA 和 cuDNN 都有官方支持。
  5. 适合容器化部署: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?是否多用户?)我可以为你定制一个一键安装脚本。

是否需要我提供一个自动安装脚本模板?

未经允许不得转载:秒懂云 » ubuntu 24.04深度学习环境搭建可以吗?