diff options
| author |  Smit Hinsu <hinsu@google.com> | 2018-07-14 13:16:58 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-14 13:20:25 -0700 |
| commit | fe7d1d9447a31562acb26aad7a9ffca60686c38a (patch) | |
| tree | 5592806f46f206bd896ab643e9ff0336d2875b33 /configure.py | |
| parent | 85aae3795775bf648d2e8baa56331f952d12e3e0 (diff) | |
Update default cuda compute capability while installing from sources to cover up to Volta
PiperOrigin-RevId: 204606836
Diffstat (limited to 'configure.py')
| -rw-r--r-- | configure.py | 2 |
1 files changed, 1 insertions, 1 deletions
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 ' |