Skip to content

Commit

Permalink
Fix Windows GPU TensorFlow Bazel builds.
Browse files Browse the repository at this point in the history
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
  • Loading branch information
Michael Case authored and tensorflower-gardener committed Apr 11, 2018
1 parent 8e1b323 commit 0073d13
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion configure.py
Original file line number Diff line number Diff line change
Expand Up @@ -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':
Expand Down

0 comments on commit 0073d13

Please sign in to comment.