Compile Tensorflow for Lambda

compile.sh

BASE_PATH=./

yum install -y wget

wget https://copr.fedorainfracloud.org/coprs/vbatts/bazel/repo/epel-7/vbatts-bazel-epel-7.repo -O /etc/yum.repos.d/vbatts-bazel-epel-7.repo

# Appear to have to run this twice to get it to run reliably.
yum install -y bazel; yum install -y bazel

if [ ! -d "$BASE_PATH/env" ]; then
  virtualenv -p python "$BASE_PATH/env"
fi

source $BASE_PATH/env/bin/activate

if [ ! -d "$BASE_PATH/_build" ]; then
    mkdir "$BASE_PATH/_build"
fi

cd _build

if [ ! -d "tensorflow" ]; then
  git clone https://github.com/tensorflow/tensorflow
fi

cd tensorflow

git reset --hard v1.4.0

PYTHON_BIN_PATH=$BASE_PATH/env/bin/python CC_OPT_FLAGS="-msse4.2 -msse4.1 -mavx" \
TF_NEED_JEMALLOC=1 TF_NEED_GCP=0 TF_NEED_HDFS=0 TF_ENABLE_XLA=0 \
TF_NEED_OPENCL=0 TF_NEED_CUDA=0 TF_NEED_VERBS=0 TF_NEED_GDR=0 TF_NEED_S3=0  TF_NEED_MPI=0 \
PYTHON_LIB_PATH="$BASE_PATH/env/lib64/python3.6/site-packages" ./configure

bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package --local_resources 2048,.5,1.0

bazel-bin/tensorflow/tools/pip_package/build_pip_package $BASE_PATH/_build/tensorflow_pkg

cd $BASE_PATH

pip install --upgrade $BASE_PATH/_build/tensorflow_pkg/tensorflow-*.whl