From 15bf01971d57fc918a90e25c9382edbba35cffa1 Mon Sep 17 00:00:00 2001 From: Vivek Panyam Date: Wed, 25 Nov 2020 16:16:07 -0500 Subject: [PATCH] Always build libtensorflow in a manylinux2010 compatible way --- tensorflow/tools/ci_build/builds/libtensorflow.sh | 2 ++ tensorflow/tools/ci_build/linux/libtensorflow_docker.sh | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/tensorflow/tools/ci_build/builds/libtensorflow.sh b/tensorflow/tools/ci_build/builds/libtensorflow.sh index bfd551f1772..aa7cbef7566 100755 --- a/tensorflow/tools/ci_build/builds/libtensorflow.sh +++ b/tensorflow/tools/ci_build/builds/libtensorflow.sh @@ -57,6 +57,8 @@ function build_libtensorflow_tarball() { BAZEL_OPTS="${BAZEL_OPTS} --config=cuda --crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda11:toolchain" export TF_NEED_ROCM=0 export TF_CUDA_COMPUTE_CAPABILITIES="sm_35,sm_50,sm_60,sm_70,sm_75,compute_80" + else + BAZEL_OPTS="${BAZEL_OPTS} --crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010:toolchain" fi bazel clean --expunge yes "" | ./configure diff --git a/tensorflow/tools/ci_build/linux/libtensorflow_docker.sh b/tensorflow/tools/ci_build/linux/libtensorflow_docker.sh index 7b2ba29de8c..4fd671cb6d4 100755 --- a/tensorflow/tools/ci_build/linux/libtensorflow_docker.sh +++ b/tensorflow/tools/ci_build/linux/libtensorflow_docker.sh @@ -31,7 +31,7 @@ DOCKER_CONTEXT_PATH="$(realpath ${SCRIPT_DIR}/..)" ROOT_DIR="$(realpath ${SCRIPT_DIR}/../../../../)" DOCKER_IMAGE="tf-libtensorflow-cpu" -DOCKER_FILE="Dockerfile.cpu" +DOCKER_FILE="Dockerfile.rbe.ubuntu16.04-manylinux2010" DOCKER_BINARY="docker" if [ "${TF_NEED_CUDA}" == "1" ]; then DOCKER_IMAGE="tf-tensorflow-gpu"