Skip to content

Instantly share code, notes, and snippets.



Last active Dec 26, 2017
What would you like to do?
Quick script to set up a GPU env on my desktop machine
# Thanks Fast.AI
# Based on their apache-2.0 licensed script
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 "" -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
# install Anaconda for current user
git clone ~/.pyenv
git clone ~/.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 "" -O "cudnn.tgz"
##tar -zxf cudnn.tgz
#cd cuda
#sudo cp lib64/* /usr/local/cuda/lib64/
#sudo cp include/* /usr/local/cuda/include/
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.