diff options
author | 2018-08-07 13:03:57 -0500 | |
---|---|---|
committer | 2018-08-07 13:03:57 -0500 | |
commit | b6b52565b892511720d2c10583be0da79d9483a0 (patch) | |
tree | b07ab3fc846ab2bd0c98041fbfaefd201d01c7d0 /tensorflow/tools/ci_build | |
parent | 26076b456f17a4b24f209d26dc0ea7f608913450 (diff) |
Need to export TF_NCCL_VERSION in Dockerfile
because ci_build.sh calls builds/configured which calls
configure.py before the linux gpu scripts are called
to set TF_NCCL_VERSION
Diffstat (limited to 'tensorflow/tools/ci_build')
4 files changed, 4 insertions, 10 deletions
diff --git a/tensorflow/tools/ci_build/Dockerfile.gpu.ppc64le b/tensorflow/tools/ci_build/Dockerfile.gpu.ppc64le index a404f129ab..a99d2c1af2 100644 --- a/tensorflow/tools/ci_build/Dockerfile.gpu.ppc64le +++ b/tensorflow/tools/ci_build/Dockerfile.gpu.ppc64le @@ -26,3 +26,6 @@ ENV LD_LIBRARY_PATH /usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH # Configure the build for our CUDA configuration. ENV TF_NEED_CUDA 1 ENV TF_CUDA_COMPUTE_CAPABILITIES 3.0 + +#TODO get NCCL 2 in the docker image +ENV TF_NCCL_VERSION 1 diff --git a/tensorflow/tools/ci_build/linux/ppc64le/gpu/run_cc_core.sh b/tensorflow/tools/ci_build/linux/ppc64le/gpu/run_cc_core.sh index b220676b77..728b89af28 100755 --- a/tensorflow/tools/ci_build/linux/ppc64le/gpu/run_cc_core.sh +++ b/tensorflow/tools/ci_build/linux/ppc64le/gpu/run_cc_core.sh @@ -28,9 +28,6 @@ echo "" export PYTHON_BIN_PATH=`which python3` export CC_OPT_FLAGS='-mcpu=power8 -mtune=power8' -#TODO get NCCL 2 in the docker image -export TF_NCCL_VERSION=1 - export TF_NEED_CUDA=1 export TF_CUDA_COMPUTE_CAPABILITIES=3.7 diff --git a/tensorflow/tools/ci_build/linux/ppc64le/gpu/run_py2_core.sh b/tensorflow/tools/ci_build/linux/ppc64le/gpu/run_py2_core.sh index 019b52d8a0..3798498abf 100755 --- a/tensorflow/tools/ci_build/linux/ppc64le/gpu/run_py2_core.sh +++ b/tensorflow/tools/ci_build/linux/ppc64le/gpu/run_py2_core.sh @@ -25,12 +25,9 @@ echo "Bazel will use ${N_JOBS} concurrent job(s)." echo "" # Run configure. -export PYTHON_BIN_PATH=`which python` +export PYTHON_BIN_PATH=`which python2` export CC_OPT_FLAGS='-mcpu=power8 -mtune=power8' -#TODO get NCCL 2 in the docker image -export TF_NCCL_VERSION=1 - export TF_NEED_CUDA=1 export TF_CUDA_COMPUTE_CAPABILITIES=3.7 diff --git a/tensorflow/tools/ci_build/linux/ppc64le/gpu/run_py3_core.sh b/tensorflow/tools/ci_build/linux/ppc64le/gpu/run_py3_core.sh index 7a28ee87c0..cb43384013 100755 --- a/tensorflow/tools/ci_build/linux/ppc64le/gpu/run_py3_core.sh +++ b/tensorflow/tools/ci_build/linux/ppc64le/gpu/run_py3_core.sh @@ -28,9 +28,6 @@ echo "" export PYTHON_BIN_PATH=`which python3` export CC_OPT_FLAGS='-mcpu=power8 -mtune=power8' -#TODO get NCCL 2 in the docker image -export TF_NCCL_VERSION=1 - export TF_NEED_CUDA=1 export TF_CUDA_COMPUTE_CAPABILITIES=3.7 |