diff options
author | Michael Case <mikecase@google.com> | 2018-04-11 09:34:44 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-11 09:41:42 -0700 |
commit | 0073d1375add58b0493449c356af76aa33455f7d (patch) | |
tree | 61f8957a45d0b74dc63a5b5b15e8c363c9893439 | |
parent | 8e1b323be4b5d56d531b2d5ee7a1fc573a2a0b5f (diff) |
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
-rw-r--r-- | configure.py | 3 |
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': |