Ubuntu 16.04 CUDA Install Script

install_cuda.sh

#!/bin/bash
cd /tmp
wget https://developer.nvidia.com/compute/cuda/9.1/Prod/local_installers/cuda-repo-ubuntu1604-9-1-local_9.1.85-1_amd64
sudo dpkg -i cuda-repo-ubuntu1604-9-1-local_9.1.85-1_amd64
sudo apt-key add /var/cuda-repo-9-1-local/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda -y
echo "-----------------"
echo "INSTALL COMPLETE."
echo "HIT ANY KEY"
echo "TO RESTART."
echo "-----------------"
read -n 1
sudo reboot