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