From fe7d1d9447a31562acb26aad7a9ffca60686c38a Mon Sep 17 00:00:00 2001
From: Smit Hinsu <hinsu@google.com>
Date: Sat, 14 Jul 2018 13:16:58 -0700
Subject: Update default cuda compute capability while installing from sources
 to cover up to Volta

PiperOrigin-RevId: 204606836
---
 configure.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'configure.py')

diff --git a/configure.py b/configure.py
index d411214817..df6259778e 100644
--- a/configure.py
+++ b/configure.py
@@ -36,7 +36,7 @@ except ImportError:
 _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 '
-- 
cgit v1.2.3