forked from RyanZotti/Self-Driving-Car
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- 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.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 |