| commit | 28fc9cccd176a113b9f0401d0c569ca1b9d207c3 | [log] [tgz] |
|---|---|---|
| author | A. Unique TensorFlower <gardener@tensorflow.org> | Wed May 01 14:17:54 2019 -0700 |
| committer | TensorFlower Gardener <gardener@tensorflow.org> | Wed May 01 15:59:23 2019 -0700 |
| tree | fa87d688fb6bcb3c624eaf55f6bf544fe7e172d2 | |
| parent | fdc03cc4459ab2abf37552d444f0c69d3026e750 [diff] [blame] |
Fix ./configure trying to find TensorRT when it shouldn't. PiperOrigin-RevId: 246197213
diff --git a/configure.py b/configure.py index b9890d3..b564fda 100644 --- a/configure.py +++ b/configure.py
@@ -1325,7 +1325,7 @@ 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')