Scripts to install TensorFlow on an AWS EC2 GPU Instance
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
README.md
toa_part_0of2.sh
toa_part_1of2.sh
toa_part_2of2.sh

README.md

tensorflow-on-aws

Scripts to install TensorFlow on an AWS EC2 GPU Instance

These scripts are based on the excellent guide by @ramhiser available here

The first part takes about 10 mins and the second part takes about 30 mins to finish

Note about cuDNN

This script requires an Nvidia Accelerated Computing Developer Program account. If you do not have an account already, register for an account at: https://developer.nvidia.com Once accepted (typically takes about a day), download cuDNN from: https://developer.nvidia.com/rdp/assets/cudnn-65-linux-v2-asset Host this file on a private location and provide the URL when the script runs.

Note about versions

This script is for the following specific versions:

Name Version
TensorFlow 0.9.0
CUDA 7.0
cuDNN 7.0
Java 8
Bazel 0.2.3
Python 2.7.6

Usage

  • Request an AWS EC2 (spot) instance with the following settings:
    • AMI: Ubuntu Server 14.04 LTS (HVM), SSD Volume Type - ami-fce3c696
    • Type: g2.2xlarge
    • Storage: 16 GB (or higher)
  • SSH into the instance
  • Download part 0 of 3 of the script
    • wget https://raw.githubusercontent.com/jasimpson/tensorflow-on-aws/master/toa_part_0of2.sh
  • Set executable
    • chmod +x toa_part_0of2.sh
  • Execute script part 0 of 2
    • ./toa_part_0of2.sh
  • Execute script part 1 of 2
    • ./toa_part_1of2.sh
  • Enter URL to download cuDNN from
  • Reboot machine
    • sudo reboot
  • SSH back into machine
  • Execute script part 2 of 2
    • ./toa_part_2of2.sh
  • Test TensorFlow
    • $ python
    • >>> import tensorflow as tf