Deep Learning is amazing, but installing the required tools can be difficult.
This script will download and install NVIDIA drivers, CUDA, CuDNN, and a Python 3 virtual environment including Pytorch, Tensorflow, and Keras.
Format your hard drive and install a brand new, clean fresh copy of Ubuntu Server 16.04
git clone https://github.com/lwneal/install-keras && cd install-keras && sudo ./install.sh
If you see an error message, check the Issues page.
Have fun with deep learning. Check out the PyTorch Examples.
If anything goes wrong with your CUDA or CuDNN installations, I find the following helpful:
sudo nvidia-uninstall
sudo apt-get remove --purge nvidia-*
sudo rm -r /usr/local/cuda*
Then try reinstalling everything again.
This script works if you are installing CUDA and related tools directly, on a machine that you have root access to.
If you have different requirements, you might be interested in Docker containers, pre-built AMI images, or services like FloydHub.