Installing TensorFlow on AWS GPU (TensorFlow 0.10, python 2.7). This script will also install CUDA toolkit 7.5 and CuDNN v5.

Python2.7_setup_aws_tensorflow.bash

#!/bin/bash

# stop on error
set -e

# install the required packages
sudo apt-get update && sudo apt-get -y upgrade
sudo apt-get -y install linux-headers-$(uname -r) linux-image-extra-`uname -r`

# install cuda 7.5
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-7-5_7.5-18_amd64.deb
sudo dpkg -i cuda-7-5_7.5-18_amd64.deb
rm cuda-7-5_7.5-18_amd64.deb
sudo apt-get update
sudo apt-get install -y cuda

# get cudnn
# Register and download the cuDNN v5 from https://developer.nvidia.com/rdp/cudnn-download
CUDNN_FILE=cudnn-7.5-linux-x64-v5.0-ga.tar
wget https://developer.nvidia.com/rdp/assets/${CUDNN_FILE}
tar xvf ${CUDNN_FILE}
rm ${CUDNN_FILE}
sudo cp cuda/include/cudnn.h /usr/local/cuda/include # move library files to /usr/local/cuda
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
rm -rf cuda

# set the appropriate library path
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
# https://www.continuum.io/downloads
wget https://repo.continuum.io/archive/Anaconda2-4.1.1-Linux-x86_64.sh
bash Anaconda2-4.1.1-Linux-x86_64.sh -b -p /home/ubuntu/anaconda2
rm Anaconda2-4.1.1-Linux-x86_64.sh
echo 'export PATH="/home/ubuntu/anaconda2/bin:$PATH"' >> ~/.bashrc

# Ubuntu/Linux 64-bit, GPU enabled, Python 2.7 tensorflow
# https://www.tensorflow.org/versions/r0.10/get_started/os_setup.html
export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.10.0-cp27-none-linux_x86_64.whl

/home/ubuntu/anaconda2/bin/pip install $TF_BINARY_URL

# install monitoring programs
sudo wget https://git.io/gpustat.py -O /usr/local/bin/gpustat
sudo chmod +x /usr/local/bin/gpustat
sudo nvidia-smi daemon
sudo apt-get -y install htop

# update terminal to latest update of .bashrc file
source .bashrc
exec bash

# If you have any questions, please feel free to email me
# email: [email protected]