Automatically install Nvidia drivers and Nvidia runtime docker in Azure

install_nvidia_drivers_and_docker_azure.sh

# Install nvidia-drivers
add-apt-repository -y ppa:graphics-drivers/ppa && \
apt-get update && \
apt-get install -y nvidia-415 && \
# Install the latest docker-ce
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | apt-key add - && \
apt-key fingerprint 0EBFCD88 && \
add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable" && \
apt-get update && \
apt-get install -y docker-ce && \
service docker restart && \
# Install nvidia docker runtime
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | apt-key add - && \
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)  && \
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | tee /etc/apt/sources.list.d/nvidia-docker.list && \
apt-get update && \
apt-get install -y  -o Dpkg::Options::="--force-confnew" nvidia-container-runtime  nvidia-docker2  && \
service docker restart