Docker containers with NVIDIA GPUs using nvidia-docker

These are the installation instructions for Ubuntu:

• Follow the docker-ce installation from (removing docker.io):
  https://docs.docker.com/install/linux/docker-ce/ubuntu
• Install nvidia-docker:
  https://github.com/NVIDIA/nvidia-docker

Test the installation running

$ docker run --runtime=nvidia --rm nvidia/cuda:latest nvidia-smi

 or the equivalent nvidia-docker wrapper call

$ nvidia-docker run --rm nvidia/cuda:latest nvidia-smi

The wrapper allows to easily select which of the host's GPUs are visible from the docker instance, in this example only the first two GPUs

$ NV_GPU=0,1 nvidia-docker run --rm -v /home/foo/data:/data -it nvidia/cuda:latest bash