nvidia-docker2

https://github.com/NVIDIA/nvidia-docker/wiki/Installation-(version-2.0)#prerequisites

https://www.nvidia.com.tw/object/docker-container-tw.html

Prerequisites

The list of prerequisites for running nvidia-docker 2.0 is described below.
For information on how to install Docker for your Linux distribution, please refer to the Docker documentation.

1. GNU/Linux x86_64 with kernel version > 3.10
2. Docker >= 1.12
3. NVIDIA GPU with Architecture > Fermi (2.1)
4. NVIDIA drivers ~= 361.93 (untested on older versions)

Your driver version might limit your CUDA capabilities (see CUDA requirements)

Removing nvidia-docker 1.0

Version 1.0 of the nvidia-docker package must be cleanly removed before continuing.
You must stop and remove all containers started with nvidia-docker 1.0.

Ubuntu distributions

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

Installing version 2.0

 

sudo apt-get install nvidia-docker2
sudo pkill -SIGHUP dockerd

Basic usage

nvidia-docker registers a new container runtime to the Docker daemon.
You must select the nvidia runtime when using docker run:
docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi