安装nvidia-docker

发表于 更新于 分类于 阅读次数： 本文字数： 697 阅读时长 ≈ 1 分钟

测试环境为: ubuntu == 18.04

官方文档: https://github.com/nvidia/nvidia-docker/wiki/Installation-(version-2.0)

ubuntu18.04版本需要安装的是V2.0版本, 网上文章大多是V1.0的, 在这个版本的ubutnu无法安装

安装命令: 

1
2
3
4
5
6
7
8
9
10
11
12
# 添加源
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
  sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
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-docker2
# 重启
sudo systemctl daemon-reload
sudo systemctl restart docker

启动镜像

1
nvidia-docker run -p 8888:8888   --dns 8.8.8.8 --dns 8.8.4.4 gcr.io/kubeflow-images-public/tensorflow-1.12.0-notebook-gpu:v0.5.0

—dns 8.8.8.8 —dns 8.8.4.4指定公网DNS, 不然容器里面无法访问网络