# Build TensorFlow Docker images for remote build
#
# Usage:
-# remote_docker_build.sh -c # docker image for cpu build
-# remote_docker_build.sh -g # docker image for gpu build
-
+# ci_rbe_docker_build.sh -c # docker image for cpu build
+# ci_rbe_docker_build.sh -g # docker image for gpu build
function main {
- publish=true
cpu_build=false
gpu_build=false
- publish=true
+ publish=false
script_dir=$(dirname "$(readlink -f "$0")")
cd $script_dir
- trap cleanup_on_finish EXIT
-
set_script_flags $@
- build_base_image
-
build_tf_image
if [ "$publish" = true ] ; then
c)
cpu_build=true
;;
- f)
- base_image_build_script=$OPTARG
- ;;
g)
gpu_build=true
;;
h)
print_usage
;;
- n)
- publish=false
+ p)
+ publish=true
;;
*)
print_usage "ERROR: unknown option"
echo "Usage: $(basename $0) -c | -g [options]"
echo " -c build image for CPU build (base image debian8-clang)"
echo " -g build image for GPU build (base image nvidia-clang)"
- echo " -f the script which build the {debian8,nvidia}-clang base image"
echo "[option] is one of"
echo " -n not publish the locally-built image to GCR;"
echo " the build process will publish image to GCR by default"
exit 1
}
-
-# Build nvidia-cuba-clang base image for GPU image.
-# For CPU the `clang-debian8` from Cloud Launcher will be used directly:
-# https://console.cloud.google.com/launcher/details/google/clang-debian8?filter=category:developer-tools&q=clang
-function build_base_image {
- if [ "$gpu_build" = true ] ; then
- base_image="nvidia-cuda"
- # Run a 2-stage build for clang base image, see
- # https://github.com/llvm-mirror/llvm/blob/master/docs/Docker.rst
- $base_image_build_script \
- --source $base_image \
- --branch branches/google/stable \
- --docker-repository ${base_image}-clang --docker-tag "latest" \
- -p clang -i stage2-install-clang -i stage2-install-clang-headers \
- -- \
- -DLLVM_TARGETS_TO_BUILD=Native -DCMAKE_BUILD_TYPE=Release \
- -DBOOTSTRAP_CMAKE_BUILD_TYPE=Release \
- -DCLANG_ENABLE_BOOTSTRAP=ON \
- -DCLANG_BOOTSTRAP_TARGETS="install-clang;install-clang-headers"
- fi
-}
-
-
function build_tf_image {
if [ "$cpu_build" = true ] ; then
- dockerfile="Dockerfile.cpu"
- tf_image="tensorflow-remote"
+ dockerfile="Dockerfile.rbe.cpu"
+ tf_image="tensorflow-rbe-cpu"
else
- dockerfile="Dockerfile.gpu"
- tf_image="tensorflow-remote-gpu"
+ dockerfile="Dockerfile.rbe.gpu"
+ tf_image="tensorflow-rbe-gpu"
fi
docker build -f $dockerfile -t $tf_image .
}
-
function publish_tf_image {
gcr_tf_image="gcr.io/tensorflow/${tf_image}"
docker tag $tf_image $gcr_tf_image
gcloud docker -- push $gcr_tf_image
}
-
-function cleanup_on_finish {
- cd $script_dir
- rm -rf $llvm_docker_src
- docker rmi -f ${base_image}-clang ${base_image}-clang-build
-}
-
-
main $@
--- /dev/null
+#!/usr/bin/env bash
+# Copyright 2015 The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+
+set -e
+
+if [ ! -f /usr/bin/x86_64-linux-gnu-gcc ]; then
+ ln -s /usr/local/bin/clang /usr/bin/x86_64-linux-gnu-gcc
+fi
+
+pip2 install -U pip
+pip3 install -U pip
+pip2 install -U setuptools
+pip3 install -U setuptools
+
+# The rest of the pip packages will be installed in
+# `install_pip_packages.sh`
projects / platform / upstream / tensorflow.git / commitdiff