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

Change the default TensorRT version to 6.

Browse Source 
PiperOrigin-RevId: 276359913
Change-Id: I1982e83d5d214375cdf153a093a003c68bdcedea
This commit is contained in:
Guangda Lai 2019-10-23 15:07:12 -07:00 committed by TensorFlower Gardener
parent a844d29343
commit faa93acb0c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
configure.py
View File

@ -35,7 +35,7 @@ except ImportError:
_DEFAULT_CUDA_VERSION = '10' _DEFAULT_CUDA_VERSION = '10'
_DEFAULT_CUDNN_VERSION = '7' _DEFAULT_CUDNN_VERSION = '7'
_DEFAULT_TENSORRT_VERSION = '5' _DEFAULT_TENSORRT_VERSION = '6'
_DEFAULT_CUDA_COMPUTE_CAPABILITIES = '3.5,7.0' _DEFAULT_CUDA_COMPUTE_CAPABILITIES = '3.5,7.0'
_TF_OPENCL_VERSION = '1.2' _TF_OPENCL_VERSION = '1.2'
@ -945,7 +945,7 @@ def set_tf_tensorrt_version(environ_cp):
ask_tensorrt_version = ( ask_tensorrt_version = (
'Please specify the TensorRT version you want to use. ' '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( tf_tensorrt_version = get_from_env_or_user_or_default(
environ_cp, 'TF_TENSORRT_VERSION', ask_tensorrt_version, environ_cp, 'TF_TENSORRT_VERSION', ask_tensorrt_version,
_DEFAULT_TENSORRT_VERSION) _DEFAULT_TENSORRT_VERSION)