Nvidia Docker

nvidia-docker.sh

# make sure you have Ubuntu 18.04 and Docker >= 1.12 installed

# nvidia drivers (Ubuntu-based distros)
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

sudo apt-get install ubuntu-drivers-common

# show list of drivers to install
sudo ubuntu-drivers devices

# install recommended drivers
sudo ubuntu-drivers autoinstall

# add nvidia repos
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

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

# test
docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi