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

Attempt to build libtensorflow GPU with the manylinux2010 toolchain.

Browse Source 
PiperOrigin-RevId: 311773671
Change-Id: I6a0a34852786fb2187ea7ad131a6e4878c84e089
This commit is contained in:
Amit Patankar 2020-05-15 11:52:10 -07:00 committed by TensorFlower Gardener
parent 9cb8d45b72
commit 362818b71e
2 changed files with 2 additions and 2 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

@ -54,7 +54,7 @@ function build_libtensorflow_tarball() {
BAZEL_OPTS="--config=opt --cxxopt=-D_GLIBCXX_USE_CXX11_ABI=0"
export CC_OPT_FLAGS="-mavx -msse4.2"
if [ "${TF_NEED_CUDA}" == "1" ]; then
BAZEL_OPTS="${BAZEL_OPTS} --config=cuda"
BAZEL_OPTS="${BAZEL_OPTS} --config=cuda --crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.1:toolchain"
export TF_NEED_ROCM=0
fi
bazel clean --expunge

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

@ -36,7 +36,7 @@ DOCKER_BINARY="docker"
if [ "${TF_NEED_CUDA}" == "1" ]; then
DOCKER_IMAGE="tf-tensorflow-gpu"
DOCKER_BINARY="nvidia-docker"
DOCKER_FILE="Dockerfile.gpu"
DOCKER_FILE="Dockerfile.rbe.cuda10.1-cudnn7-ubuntu16.04-manylinux2010"
fi
if [ "${TF_NEED_ROCM}" == "1" ]; then
DOCKER_IMAGE="tf-tensorflow-rocm"