Quick script to set up a GPU env on my desktop machine

install-gpu.sh

# Thanks Fast.AI
# Based on their apache-2.0 licensed script
#   https://github.com/fastai/courses/blob/master/setup/install-gpu.sh
# 

set -e
# This script is designed to work with ubuntu 16.04 LTS
# ensure system is updated and has basic build tools
sudo apt-get update
sudo apt-get --assume-yes upgrade
sudo apt-get --assume-yes install tmux build-essential gcc g++ make binutils
sudo apt-get --assume-yes install software-properties-common
sudo apt-get install -y linux-headers-4.13.0-21-generic linux-4.13.0-21-generic

# download and install GPU drivers
wget "http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_8.0.44-1_amd64.deb" -O "cuda-repo-ubuntu1604_8.0.44-1_amd64.deb"

sudo dpkg -i cuda-repo-ubuntu1604_8.0.44-1_amd64.deb
sudo apt-get update
sudo apt-get -y install cuda
sudo modprobe nvidia
nvidia-smi

# install Anaconda for current user
git clone https://github.com/pyenv/pyenv.git ~/.pyenv
git clone https://github.com/pyenv/pyenv-virtualenv.git ~/.pyenv/plugins/pyenv-virtualenv
echo 'export PYENV_ROOT="$HOME/.pyenv"' >> ~/.bash_profile
echo 'export PATH="$PYENV_ROOT/bin:$PATH"' >> ~/.bash_profile
echo 'eval "$(pyenv init -)"' >> ~/.bash_profile
echo 'eval "$(pyenv virtualenv-init -)"' >> ~/.bash_profile

source ~/.bash_profile

pyenv install anaconda3-5.0.1
pyenv virtualenv anaconda3-5.0.1 nn-experiments
pyenv activate nn-experiments
conda install -y tensorflow-gpu jupyter numpy scipy scikit-learn tqdm keras matplotlib
conda install -y pytorch torchvision -c pytorch

# The following are not needed! Thankfully!
# I loath nvidia and its shitty setup.
#wget "http://files.fast.ai/files/cudnn.tgz" -O "cudnn.tgz"
##tar -zxf cudnn.tgz
#cd cuda
#sudo cp lib64/* /usr/local/cuda/lib64/
#sudo cp include/* /usr/local/cuda/include/