Change the default TensorRT version to 6.
PiperOrigin-RevId: 276359913
Change-Id: I1982e83d5d214375cdf153a093a003c68bdcedea
diff --git a/configure.py b/configure.py
index 92b1bd1..46ce13e 100644
--- a/configure.py
+++ b/configure.py
@@ -35,7 +35,7 @@
_DEFAULT_CUDA_VERSION = '10'
_DEFAULT_CUDNN_VERSION = '7'
-_DEFAULT_TENSORRT_VERSION = '5'
+_DEFAULT_TENSORRT_VERSION = '6'
_DEFAULT_CUDA_COMPUTE_CAPABILITIES = '3.5,7.0'
_TF_OPENCL_VERSION = '1.2'
@@ -945,7 +945,7 @@
ask_tensorrt_version = (
'Please specify the TensorRT version you want to use. '
- '[Leave empty to default to TensorRT %s]: ') % _DEFAULT_TENSORRT_VERSION
+ '[Leave empty to default to TensorRT %s]: ') % _DEFAULT_TENSORRT_VERSION
tf_tensorrt_version = get_from_env_or_user_or_default(
environ_cp, 'TF_TENSORRT_VERSION', ask_tensorrt_version,
_DEFAULT_TENSORRT_VERSION)