diff options
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 |