Skip to content

Commit

Permalink
automated GPU setup installation
Browse files Browse the repository at this point in the history
  • Loading branch information
Ryan Zotti authored and Ryan Zotti committed Aug 28, 2016
1 parent 3f795d6 commit f76e17e
Showing 1 changed file with 51 additions and 0 deletions.
51 changes: 51 additions & 0 deletions gpu_setup.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,51 @@
# Do everything as root just to make things easier
sudo su

# Update the GPU and say 'yes' to any questions
apt-get update && apt-get -y upgrade

# Not sure what this is trying to install
apt-get -y install linux-headers-$(uname -r) linux-image-extra-`uname -r`

# Get and install CUDA
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_7.5-18_amd64.deb
dpkg -i cuda-repo-ubuntu1404_7.5-18_amd64.deb

# Upgrade/install again specifically for CUDA
apt-get update -y
apt-get install -y cuda

# Get and install CUDNN
CUDNN_FILE=cudnn-7.0-linux-x64-v4.0-prod.tgz
wget http://developer.download.nvidia.com/compute/redist/cudnn/v4/${CUDNN_FILE}
tar xvzf ${CUDNN_FILE}

cp cuda/include/cudnn.h /usr/local/cuda/include
cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

echo 'export CUDA_HOME=/usr/local/cuda
export CUDA_ROOT=/usr/local/cuda
export PATH=$PATH:$CUDA_ROOT/bin:$HOME/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CUDA_ROOT/lib64
' >> ~/.bashrc

# Install Anaconda
# Python will break if you try to put it in /root, so use /mnt/bin instead
mkdir -p /mnt/bin
wget http://repo.continuum.io/archive/Anaconda3-4.0.0-Linux-x86_64.sh
bash Anaconda3-4.0.0-Linux-x86_64.sh -b -p /mnt/bin/anaconda3
echo 'export PATH="/mnt/bin/anaconda3/bin:$PATH"' >> ~/.bashrc

# Turn on bash profile to activate the PATH variables from above
exec bash

# Download and install TensorFlow binary (way easier than using source)
export TF_BINARY_URL='https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.9.0rc0-cp35-cp35m-linux_x86_64.whl'
/mnt/bin/anaconda3/bin/pip install $TF_BINARY_URL

# Install git so you can clone your repo and then run your code
apt-get install -y git

# Get data training and validation data
wget https://s3.amazonaws.com/self-driving-car/final_processed_data_3_channels.npz

0 comments on commit f76e17e

Please sign in to comment.