commit fe7d1d9447a31562acb26aad7a9ffca60686c38a
author Smit Hinsu <hinsu@google.com> Sat Jul 14 13:16:58 2018 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Sat Jul 14 13:20:25 2018 -0700
tree 5592806f46f206bd896ab643e9ff0336d2875b33
parent 85aae3795775bf648d2e8baa56331f952d12e3e0

Update default cuda compute capability while installing from sources to cover up to Volta

PiperOrigin-RevId: 204606836

configure.py

diff --git a/configure.py b/configure.py
index d411214..df62597 100644
--- a/configure.py
+++ b/configure.py
@@ -36,7 +36,7 @@
 _DEFAULT_CUDA_VERSION = '9.0'
 _DEFAULT_CUDNN_VERSION = '7'
 _DEFAULT_NCCL_VERSION = '2.2'
-_DEFAULT_CUDA_COMPUTE_CAPABILITIES = '3.5,5.2'
+_DEFAULT_CUDA_COMPUTE_CAPABILITIES = '3.5,7.0'
 _DEFAULT_CUDA_PATH = '/usr/local/cuda'
 _DEFAULT_CUDA_PATH_LINUX = '/opt/cuda'
 _DEFAULT_CUDA_PATH_WIN = ('C:/Program Files/NVIDIA GPU Computing '