安装 Docker 和 Nvidia-docker2

2 年 ago

逸, 科

1 minute

你好，我是chlochan。我正在將系統從Ubuntu 18.04+CUDA10轉移到CUDA9.0的環境。因為有時候也需要CUDA9.0，所以我打算使用nvidia-docker2來建立CUDA9.0的環境。

安装Docker

首先安装Docker。

$ sudo apt-get update
$ sudo apt-get install apt-transport-https ca-certificates curl software-properties-common
$ curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
$ sudo apt-key fingerprint 0EBFCD88
$ sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu \
   $(lsb_release -cs) stable"
$ sudo apt-get update
$ sudo apt-get install docker-ce

我会尝试确认是否已成功安装。

$ sudo docker container run hello-world
$ sudo docker image
$ sudo docker ps -a

如果没有问题的话，将删除 Hello-World 容器和 Docker 镜像。

$ sudo docker container rm <hello-worldのコンテナID>
$ sudo docker rmi hello-world

安装nvidia-docker2。

目前，用于深度学习的GPU工具依赖于CUDA版本。例如，TensorFlow Ver1.12的编译二进制映像在运行时需要CUDA9.0库，否则会出错。因此，需要安装所需工具所需的CUDA Toolkit版本。nvidia-docker2是Docker容器运行时，它允许在每个容器中使用不同版本的CUDA Toolkit，只要主机操作系统的CUDA驱动程序保持最新，就可以极大地简化运营过程。

请根据以下 Quickstart 进行安装。
https://github.com/NVIDIA/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
$ sudo apt-get install -y nvidia-docker2
$ sudo pkill -SIGHUP dockerd

根据下面的内容，9.0-cudnn7-devel是安装了CUDA9.0全部组件的映像，我将使用它来进行操作验证。
https://hub.docker.com/r/nvidia/cuda/

$ sudo docker container run --runtime=nvidia --rm nvidia/cuda:9.0-cudnn7-devel nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_21:08:03_CDT_2017
Cuda compilation tools, release 9.0, V9.0.176

我会确认是否已经升级到V9.0.176版本。