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

Fix ./configure trying to find TensorRT when it shouldn't.

Browse Source 
PiperOrigin-RevId: 246197213
This commit is contained in:
A. Unique TensorFlower 2019-05-01 14:17:54 -07:00 committed by TensorFlower Gardener
parent fdc03cc445
commit 28fc9cccd1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configure.py
View File

@ -1325,7 +1325,7 @@ def validate_cuda_config(environ_cp):
cuda_libraries = ['cuda', 'cudnn']
if is_linux():
if environ_cp.get('TF_NEED_TENSORRT', None):
if int(environ_cp.get('TF_NEED_TENSORRT', False)):
cuda_libraries.append('tensorrt')
if environ_cp.get('TF_NCCL_VERSION', None):
cuda_libraries.append('nccl')