当前位置:首页 > 系统教程 > 正文

Ubuntu 24.04 LTS NVIDIA驱动安装教程(CUDA 12.5与Docker容器工具包保姆级指南)

前言:

在Ubuntu 24.04 LTS发布后,许多开发者需要配置高效的GPU加速环境。本文将详细介绍如何在该系统上安装 Ubuntu 24.04 NVIDIA驱动、最新的 CUDA 12.5安装 以及 Docker容器工具包,帮助你快速搭建 深度学习环境配置

第一步:更新系统并卸载旧驱动

在开始之前,我们需要确保系统软件包是最新的,并清理可能冲突的旧驱动。

    sudo apt update && sudo apt upgrade -ysudo apt autoremove --purge nvidia*  

第二步:安装 NVIDIA 显卡驱动

Ubuntu 24.04 提供了非常方便的驱动管理工具。我们推荐使用系统推荐的生产版本。

    # 查看推荐驱动版本ubuntu-drivers devices# 安装推荐驱动(通常是 nvidia-driver-550 或更高版本)sudo apt install nvidia-driver-550  

安装完成后,必须重启电脑sudo reboot。重启后输入 nvidia-smi 验证是否成功。

Ubuntu 24.04 LTS NVIDIA驱动安装教程(CUDA 12.5与Docker容器工具包保姆级指南) NVIDIA驱动  CUDA 12.5安装 Docker容器工具包 深度学习环境配置 第1张

第三步:安装 CUDA 12.5 Toolkit

前往 NVIDIA 官网获取 CUDA 12.5 的安装命令,或者直接执行以下脚本:

    wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.debsudo dpkg -i cuda-keyring_1.1-1_all.debsudo apt updatesudo apt install cuda-toolkit-12-5  

配置环境变量:编辑 ~/.bashrc,在末尾添加:

    export PATH=/usr/local/cuda-12.5/bin${PATH:+:${PATH}}export LD_LIBRARY_PATH=/usr/local/cuda-12.5/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}  

执行 source ~/.bashrc 使其生效。

第四步:安装 Docker 引擎

为了使用 Docker容器工具包,首先需要安装基础 Docker 环境。

    curl -fsSL https://get.docker.com | bash -s dockersudo usermod -aG docker $USER  

第五步:配置 NVIDIA Container Toolkit

这是让 Docker 容器能够识别并调用宿主机显卡的关键步骤。

    # 添加源curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpgcurl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list# 安装工具包sudo apt updatesudo apt install -y nvidia-container-toolkit# 配置并重启 Dockersudo nvidia-ctk runtime configure --runtime=dockersudo systemctl restart docker  

最终测试

运行以下命令,如果能看到 GPU 状态输出,说明整个环境配置成功:

    docker run --rm --gpus all nvidia/cuda:12.5.0-base-ubuntu22.04 nvidia-smi  

本文涉及的核心关键词:Ubuntu 24.04 NVIDIA驱动、CUDA 12.5安装、Docker容器工具包、深度学习环境配置。祝您开发愉快!