From 1a83c83e35bd1db84d4764d18937ce894dcc4388 Mon Sep 17 00:00:00 2001 From: Shanqing Cai Date: Mon, 6 Jun 2016 09:24:36 -0400 Subject: parameterized_docker_build includes three CUDA capabilities --- tensorflow/tools/docker/parameterized_docker_build.sh | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tensorflow/tools/docker/parameterized_docker_build.sh b/tensorflow/tools/docker/parameterized_docker_build.sh index 73fd70c604..3b35a4f3bc 100755 --- a/tensorflow/tools/docker/parameterized_docker_build.sh +++ b/tensorflow/tools/docker/parameterized_docker_build.sh @@ -179,6 +179,9 @@ if [[ "${DO_PIP_BUILD}" == "1" ]]; then export TF_BUILD_IS_OPT="OPT" export TF_BUILD_IS_PIP="PIP" + export TF_BUILD_APPEND_CI_DOCKER_EXTRA_PARAMS=\ +"-e TF_CUDA_COMPUTE_CAPABILITIES=3.0,3.5,5.2" + pushd "${SCRIPT_DIR}/../../../" rm -rf pip_test/whl && tensorflow/tools/ci_build/ci_parameterized_build.sh -- cgit v1.2.3