experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Always build libtensorflow in a manylinux2010 compatible way

Browse Source
This commit is contained in:
Vivek Panyam 2020-11-25 16:16:07 -05:00
parent 287d8116bf
commit 15bf01971d
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/tools/ci_build/builds/libtensorflow.sh
View File

@ -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

2
tensorflow/tools/ci_build/linux/libtensorflow_docker.sh
View File

@ -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"