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

Ubuntu 22.04 完美安装Docker与NVIDIA显卡驱动教程

Ubuntu 22.04 完美安装Docker与NVIDIA显卡驱动教程

从零开始配置NVIDIA Container Toolkit,实现GPU容器化

本文详细讲解在Ubuntu 22.04系统中安装Docker镜像环境、NVIDIA显卡驱动以及NVIDIA Container Toolkit的全过程。即使是小白用户,按照步骤操作也能轻松完成Docker GPU支持的配置。

Ubuntu 22.04 完美安装Docker与NVIDIA显卡驱动教程 Docker安装  NVIDIA显卡驱动 NVIDIA Container Toolkit Docker GPU支持 第1张

1. 更新系统与准备

打开终端,执行以下命令更新软件包列表并升级现有软件:

    sudo apt update && sudo apt upgrade -y  

2. 安装NVIDIA显卡驱动

NVIDIA显卡驱动是让GPU正常工作并支持CUDA的基础。Ubuntu 22.04提供了简便的自动安装方式:

    # 查看推荐的驱动版本ubuntu-drivers devices# 自动安装推荐驱动sudo ubuntu-drivers autoinstall  

安装完成后,重启系统使驱动生效:sudo reboot。重启后运行nvidia-smi验证驱动是否成功安装,若显示GPU信息则说明驱动正常。

3. 安装Docker CE

Ubuntu 22.04 Docker安装推荐使用官方仓库,以获得最新稳定版本。依次执行:

    # 卸载旧版本sudo apt remove docker docker-engine docker.io containerd runc# 安装依赖sudo apt install ca-certificates curl gnupg lsb-release# 添加Docker官方GPG密钥curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo gpg --dearmor -o /usr/share/keyrings/docker-archive-keyring.gpg# 设置稳定版仓库echo "deb [arch=$(dpkg --print-architecture) signed-by=/usr/share/keyrings/docker-archive-keyring.gpg] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable" | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null# 安装Docker引擎sudo apt updatesudo apt install docker-ce docker-ce-cli containerd.io  

安装后启动Docker并设置开机自启:sudo systemctl enable docker --now。验证Docker安装:docker --version

4. 安装NVIDIA Container Toolkit

NVIDIA Container Toolkit允许Docker容器访问宿主机的GPU资源。首先添加NVIDIA的仓库:

    distribution=$(. /etc/os-release;echo $ID$VERSION_ID)curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list  

然后安装nvidia-docker2并重启Docker:

    sudo apt updatesudo apt install nvidia-docker2sudo systemctl restart docker  

5. 验证GPU容器运行

最后,运行一个CUDA测试容器,验证Docker GPU支持是否生效:

    sudo docker run --gpus all nvidia/cuda:11.0-base nvidia-smi  

如果输出与宿主机nvidia-smi类似的信息,说明容器成功调用GPU,整个环境配置完成!

至此,您已成功在Ubuntu 22.04上搭建了支持GPU的Docker环境,可以拉取各种Docker镜像进行深度学习或科学计算了。