aws-pytorch-anaconda-cuda.sh

#!/bin/bash

HOSTNAME=$1

# Set up environment
git clone http://github.com/spro/dotfiles
cd dotfiles && ./bin/install-dotfiles
cd ~
sudo hostname $HOSTNAME
source ~/.bash_profile

# Set up CUDA
wget http://us.download.nvidia.com/tesla/375.51/nvidia-driver-local-repo-ubuntu1604_375.51-1_amd64.deb
sudo dpkg -i nvidia-driver-local-repo-ubuntu1604_375.51-1_amd64.deb
sudo apt-get update
sudo apt-get -y install cuda-drivers

# Install Anaconda
curl -O https://repo.continuum.io/archive/Anaconda3-4.3.1-Linux-x86_64.sh
bash Anaconda3-4.3.1-Linux-x86_64.sh -b
echo "PATH=\$HOME/anaconda3/bin:\$PATH" >> ~/.paths
source ~/.paths

# Install PyTorch
conda install pytorch torchvision cuda80 -c soumith
pip install sconce visdom

# Reboot to load CUDA drivers
sudo reboot