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

2
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'