Ubuntu18.04中安装带NVIDIA的Docker环境

发表于 · 归类于代码 · 3 分钟 · 报告错误

环境准备

  1. 卸载旧版本的Docker
sudo apt-get remove docker docker-engine docker.io containerd runc
  1. 安装需要的依赖包
sudo apt-get install -y \
    apt-transport-https \
    ca-certificates \
    curl \
    gnupg-agent \
    software-properties-common
  1. 加入 Docker 官方的 GPG Key 到本地系统中
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
# 验证是否拥有指定的密钥
sudo apt-key fingerprint 0EBFCD88
  1. 加入 Docker 官方源到 APT 配置文件中
sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu bionic stable"

安装Docker

  1. 更新apt程序包索引，
sudo apt update
  1. 安装指定版本的Docker
# 查询仓库中可用的Dokcer版本
sudo apt-cache madison docker-ce

# 安装指定的版本的Docker Engine 和容器 19.03.13~3-0~ubuntu
sudo apt-get install docker-ce docker-ce-cli containerd.io
  1. 测试Docker是否安装成功
sudo docker run --rm hello-world

安装 NVIDIA Container Toolkit

  1. 配置repo并安装
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


sudo apt-get update

sudo apt-get install nvidia-container-toolkit


sudo systemctl restart docker
  1. 测试 NVIDIA Container Toolkit
sudo docker run --gpus all --rm nvidia/cuda nvidia-smi

设定用户权限

因为在安装 Docker 是，自动在系统中建了一个名叫: docker 的组（group），该组的用户有权限执行 Docker 相关的命令。所以我们只需要把当前用户加入到 docker 组中即可。执行以下命令：

sudo usermod -aG docker zhaoliu
☰