diff options
author | Shanqing Cai <cais@google.com> | 2016-06-06 09:24:36 -0400 |
---|---|---|
committer | Shanqing Cai <cais@google.com> | 2016-06-06 09:24:36 -0400 |
commit | 1a83c83e35bd1db84d4764d18937ce894dcc4388 (patch) | |
tree | ae36384e36372f128a92dde21598d79a6ff2817f | |
parent | a3a6cbbe9e40211dc54f4555351ed515b89fe2a2 (diff) |
parameterized_docker_build includes three CUDA capabilities
-rwxr-xr-x | tensorflow/tools/docker/parameterized_docker_build.sh | 3 |
1 files changed, 3 insertions, 0 deletions
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 |