【Docker】Ubuntu16.04安装nvidia-docker2

官方说随着docker19.03的发布，不赞成使用nvidia-docker2软件包，因为Docker运行时中现在已将NVIDIA GPU作为设备本地支持。但是折腾了几天，安装了docker19.03，也用不起NVIDIA GPU，所以这里还是退而求其次，安装低版本的docker，并且安装nvidia-docker2，先用起来再说，后续再研究。

---

环境：

Ubuntu 16.04 LTS

NVIDIA driver 390.87

Docker 18.09.7

nvidia-docker2 2.0.3

nvidia-container-runtime 2.0.0

---

先卸载nvidia-docker

sudo docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
sudo apt-get purge nvidia-docker

 

一、添加软件包存储库

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

二、查看可安装版本

apt-cache madison nvidia-docker2 nvidia-container-runtime

选择和已安装的docker配套的版本。

三、安装nvidia-docker2

sudo apt-get install -y nvidia-docker2=2.0.3+docker18.09.7-3 nvidia-container-runtime=2.0.0+docker18.09.7-3

四、测试

docker run --runtime nvidia nvidia/cuda:9.0-base nvidia-smi

如果没有镜像会自动下载。

说明已经调用到GPU了。

---

安装过程遇到的两个错误请参考链接：

1. 【Docker】nvdia-docker2 docker-ce （= 5 18.09.7~3-0~ubuntu-xenial）but 5 19.03.4~3-0~ubuntu-xenial is to

2. 【Docker】nvidia-container-runtime Depends nvidia-container-runtime-hook