aboutsummaryrefslogtreecommitdiffhomepage
path: root/configure.py
diff options
context:
space:
mode:
Diffstat (limited to 'configure.py')
-rw-r--r--configure.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/configure.py b/configure.py
index 81d5ad77ee..8fb8979111 100644
--- a/configure.py
+++ b/configure.py
@@ -1516,7 +1516,8 @@ def main():
set_tf_cudnn_version(environ_cp)
if is_linux():
set_tf_tensorrt_install_path(environ_cp)
- set_tf_nccl_install_path(environ_cp)
+ set_tf_nccl_install_path(environ_cp)
+
set_tf_cuda_compute_capabilities(environ_cp)
if 'LD_LIBRARY_PATH' in environ_cp and environ_cp.get(
'LD_LIBRARY_PATH') != '1':