当前位置: 首页 > 工具软件 > nvidia-docker > 使用案例 >

Ubuntu20.04 完美版极速且完整(官方版) 安装docker和可使用GPU的官方最新推荐版nvidia-docker2

宋涵忍
2023-12-01

Ubuntu20.04 安装docker 和可以使用宿主机GPU的nvidia-docker2

开始

设置存储库

  1. 更新软件包索引并安装软件包以允许使用 基于 HTTPS 的存储库
sudo apt-get update
sudo apt-get install ca-certificates curl gnupg lsb-release
  1. 添加 Docker 的官方 GPG 密钥
sudo mkdir -p /etc/apt/keyrings
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo gpg --dearmor -o /etc/apt/keyrings/docker.gpg
  1. 使用以下命令设置存储库
echo "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable" | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null

安装 Docker 引擎

  1. 更新包索引
sudo apt-get update
  1. 安装 Docker Engine、containerd 和 Docker Compose
sudo apt-get install docker-ce docker-ce-cli containerd.io docker-compose-plugin
  1. 通过运行映像验证 Docker 引擎安装是否成功
sudo docker run hello-world

以非 root 用户身份管理 Docker

  1. 创建组
sudo groupadd docker
  1. 将您的用户添加到组中
sudo usermod -aG docker $USER
  1. 激活对组的更改
newgrp docker
  1. 验证是否可以以非root用户身份管理Docker
docker run hello-world

将 Docker 配置为使用 systemd启动

sudo systemctl enable docker.service
sudo systemctl enable containerd.service

测试systemctl 配置是否成功

sudo systemctl status docker

至此ubuntu20.04安装 docker 完成。

接下来开始配置安装nvidia-docker2

安装nvidia-docker2的先决条件

  1. GNU/Linux x86_64内核版本 > 3.10
  2. Docker >= 19.03(推荐,但某些发行版可能包含旧版本的 Docker。支持的最低版本为 1.12)
  3. NVIDIA GPU 與 Architecture >= Kepler(或計算能力 3.0)
  4. NVIDIA Linux 驱动程序 >= 418.81.07(请注意,不支持较旧的驱动程序版本或分支。

满足条件后可以开始安装nvidia-docker2

  1. 设置软件包存储库和 GPG 密钥
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) && curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg && curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.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
  1. 更新包列表后安装包
sudo apt-get update
  1. 安装nvidia-docker2
sudo apt-get install -y nvidia-docker2
  1. 重启docker进程
sudo systemctl restart docker
  1. 测试nvidia-docker2是否安装成功
sudo docker run --rm --gpus all nvidia/cuda:11.6.2-base-ubuntu20.04 nvidia-smi

控制台会输出

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 450.51.06    Driver Version: 450.51.06    CUDA Version: 11.0     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  Tesla T4            On   | 00000000:00:1E.0 Off |                    0 |
| N/A   34C    P8     9W /  70W |      0MiB / 15109MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+
 类似资料: