install_nvidia_driver.sh

#! /bin/bash

# nvidia-370 seems to work
# Adding user to docker sudo (docker required)
sudo usermod -aG docker $(whoami)

# Nvidia pre-requirements
sudo apt-get update && sudo apt-get upgrade
sudo apt-get install -y linux-headers-$(uname -r)
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install -y nvidia-370
sudo apt-get install -y nvidia-modprobe

# Nvidia docker
wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.0-rc.3/nvidia-docker_1.0.0.rc.3-1_amd64.deb

sudo dpkg -i /tmp/nvidia-docker*.deb && rm /tmp/nvidia-docker*.deb