setup tensorflow on aws

setup-tensorflow.sh

#!/bin/bash
set -o nounset
set -o errexit

git clone https://github.com/tensorflow/tensorflow

sudo apt-get install -y openjdk-8-jdk libcurl4-openssl-dev python3-numpy swig python3-dev python3-pip python3-setuptools zlib1g-dev

echo "deb http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list
curl https://storage.googleapis.com/bazel-apt/doc/apt-key.pub.gpg | sudo apt-key add -

sudo apt-get update && sudo apt-get install -y bazel

cd tensorflow

PYTHON_BIN_PATH=/usr/bin/python3 \
TF_NEED_GCP=1 \
TF_NEED_CUDA=1 \
GCC_HOST_COMPILER_PATH=/usr/bin/gcc \
TF_CUDA_VERSION=8.0 \
CUDA_TOOLKIT_PATH=/usr/local/cuda \
TF_CUDNN_VERSION=5 \
CUDNN_INSTALL_PATH=/usr/local/cuda \
TF_CUDA_COMPUTE_CAPABILITIES=3.0 ./configure

bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package

bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg

cp -R /tmp/tensorflow_pkg/ .

sudo pip3 install /tmp/tensorflow_pkg/tensorflow-0.9.0-py3-none-any.whl

cd ~