From 0cc518ee98d4caa154f8a7530cb971c00c610905 Mon Sep 17 00:00:00 2001 From: Michael Case Date: Wed, 11 Apr 2018 09:34:44 -0700 Subject: [PATCH] Fix Windows GPU TensorFlow Bazel builds. The configure.py script will error out on Windows GPU builds due to NCCL attempted to be configured (and is currently Linux only). PiperOrigin-RevId: 192461362 --- configure.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/configure.py b/configure.py index 81d5ad77ee4..8fb89791116 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':