1#!/usr/bin/env bash 2 3# Copyright 2019 The TensorFlow Authors. All Rights Reserved. 4# 5# Licensed under the Apache License, Version 2.0 (the "License"); 6# you may not use this file except in compliance with the License. 7# You may obtain a copy of the License at 8# 9# http://www.apache.org/licenses/LICENSE-2.0 10# 11# Unless required by applicable law or agreed to in writing, software 12# distributed under the License is distributed on an "AS IS" BASIS, 13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14# See the License for the specific language governing permissions and 15# limitations under the License. 16# ============================================================================ 17 18 19 20# Download and build TensorFlow. 21set -euxo pipefail 22git clone --branch=master --depth=1 https://github.com/tensorflow/tensorflow.git /tensorflow 23cd /tensorflow 24 25ln -s $(which ${PYTHON}) /usr/local/bin/python 26 27# Build TensorFlow with support for Intel(R) MKL-DNN 28yes "" | ${PYTHON} configure.py && \ 29 bazel build -c opt --config=mkl --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" \ 30 tensorflow/tools/pip_package:build_pip_package && \ 31 bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/pip && \ 32 pip --no-cache-dir install --upgrade /tmp/pip/tensorflow-*.whl && \ 33 rm -rf /tmp/pip && \ 34 rm -rf /root/.cache 35 36 37# download and build Horovod 38git clone --recursive https://github.com/uber/horovod.git 39cd horovod 40# export environment 41export HOROVOD_WITHOUT_PYTORCH=1 42export HOROVOD_WITH_TENSORFLOW=1 43python setup.py sdist 44pip --no-cache-dir install --upgrade sdist/horovod*.tar.gz && \ 45 rm -rf sdist && \ 46 rm -rf /root/.cache 47