diff options
author | Anna R <annarev@google.com> | 2018-09-24 19:19:06 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-24 19:24:56 -0700 |
commit | 720594142a51c2676d086ba00705d95002474687 (patch) | |
tree | 3b99282eb59b93e6a2e2aded07991b3aa46119bf /tensorflow/tools/docker | |
parent | e78f5226b94e6bdb9c204351791198488d38d403 (diff) |
Internal change.
PiperOrigin-RevId: 214371906
Diffstat (limited to 'tensorflow/tools/docker')
-rwxr-xr-x | tensorflow/tools/docker/parameterized_docker_build.sh | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/tools/docker/parameterized_docker_build.sh b/tensorflow/tools/docker/parameterized_docker_build.sh index 448a3a7647..570aa8278c 100755 --- a/tensorflow/tools/docker/parameterized_docker_build.sh +++ b/tensorflow/tools/docker/parameterized_docker_build.sh @@ -244,7 +244,7 @@ if [[ "${TF_DOCKER_BUILD_IS_DEVEL}" == "no" ]]; then if [[ "${TF_DOCKER_BUILD_TYPE}" == "gpu" ]]; then export TF_BUILD_APPEND_CI_DOCKER_EXTRA_PARAMS=\ - "${TF_BUILD_APPEND_CI_DOCKER_EXTRA_PARAMS} -e TF_CUDA_COMPUTE_CAPABILITIES=3.0,3.5,5.2" + "${TF_BUILD_APPEND_CI_DOCKER_EXTRA_PARAMS} -e TF_CUDA_COMPUTE_CAPABILITIES=3.0,3.5,5.2,6.0" fi pushd "${SCRIPT_DIR}/../../../" |