install_nvidia_and_docker.sh

#!/bin/bash
set -eu -o pipefail
# SRC: https://github.com/nvidia/nvidia-docker/wiki/Installation-(version-2.0)
distribution=$(. /etc/os-release;echo ${ID}${VERSION_ID})
cudaversion=$(curl --silent https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/  | grep cuda-repo-ubuntu1604 | tail -1 | sed -e s/[\'\"]/\'/g | sed -e s/".*href='"//g | sed -e s/"'>.*"//g)

sudo apt-get -q update
sudo apt-get -q install -y update-motd
sudo apt-get -q -y dist-upgrade
sudo apt-get -q -y autoremove

sudo apt-get -q install -y build-essential update-motd
sudo apt-get -q install -y linux-headers-$(uname -r)

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/${cudaversion} -O ${cudaversion}
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
sudo dpkg -i ${cudaversion}

sudo apt-get -q -y purge docker docker*
sudo apt-get -q install -y apt-transport-https ca-certificates curl software-properties-common

curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey |   sudo apt-key add -

sudo add-apt-repository  "deb [arch=amd64] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable"
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 -q update
sudo apt-get -q -y install cuda nvidia-docker2
sudo usermod -aG docker ubuntu
sudo service docker restart
sudo docker pull bash:latest
sudo docker pull nvidia/cuda:latest
sudo docker pull missinglinkai/ml-rm-docker:latest
sudo docker pull missinglinkai/git-lfs:latest
sudo docker pull missinglinkai/missinglink:latest