experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Upgrade default TF cuda version to cuda 10.

Browse Source 
PiperOrigin-RevId: 227180147
This commit is contained in:
Gunhan Gulsoy 2018-12-28 16:32:01 -08:00
parent ab7929f160
commit 18f42ee4b6
1 changed files with 1 additions and 1 deletions

2
configure.py
View File

@ -33,7 +33,7 @@ except ImportError:
from distutils.spawn import find_executable as which
# pylint: enable=g-import-not-at-top
_DEFAULT_CUDA_VERSION = '9.0'
_DEFAULT_CUDA_VERSION = '10.0'
_DEFAULT_CUDNN_VERSION = '7'
_DEFAULT_CUDA_COMPUTE_CAPABILITIES = '3.5,7.0'
_DEFAULT_CUDA_PATH = '/usr/local/cuda'