Install docker and nvidia-docker

install_gpu_docker.sh

# Disco!
sudo apt-get update
sudo apt-get install linux-image-extra-$(uname -r) linux-image-extra-virtual -y
sudo apt-get install apt-transport-https ca-certificates curl software-properties-common -y
sudo apt-key fingerprint 0EBFCD88
sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable" -y
sudo apt-get update
sudo apt-get install docker.io -y
wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker_1.0.1-1_amd64.deb
sudo dpkg --ignore-depends=docker-ce -i /tmp/nvidia-docker*.deb && rm /tmp/nvidia-docker*.deb
sudo apt-get install nvidia-modprobe -y
sudo apt-add-repository multiverse 
sudo apt update
sudo apt install nvidia-modprobe -y
#sudo apt-get install -y nvidia-367
sudo nvidia-docker run --rm nvidia/cuda nvidia-smi