Update the TF CUDA version to 10.1
PiperOrigin-RevId: 280237880
Change-Id: Idb9ea64c73746e18b372f3c3a0e56b8b64d53b06
diff --git a/configure.py b/configure.py
index f1a2a8a..88445ee 100644
--- a/configure.py
+++ b/configure.py
@@ -33,7 +33,7 @@
from distutils.spawn import find_executable as which
# pylint: enable=g-import-not-at-top
-_DEFAULT_CUDA_VERSION = '10'
+_DEFAULT_CUDA_VERSION = '10.1'
_DEFAULT_CUDNN_VERSION = '7'
_DEFAULT_TENSORRT_VERSION = '6'
_DEFAULT_CUDA_COMPUTE_CAPABILITIES = '3.5,7.0'