OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-07 12:20:24 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

When GPU is configured, do not require --config=cuda.

Browse Source 
Also fix indentation in configure.

PiperOrigin-RevId: 158232959
This commit is contained in:
Gunhan Gulsoy 2017-06-06 22:34:30 -07:00 committed by TensorFlower Gardener
parent 8ad6a036e5
commit 3cf88d390c
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
configure vendored
View File

@ -670,6 +670,12 @@ if is_windows; then
write_action_env_to_bazelrc "CUDA_PATH" "$CUDA_PATH"
write_action_env_to_bazelrc "CUDA_COMPUTE_CAPABILITIES" "$CUDA_COMPUTE_CAPABILITIES"
write_action_env_to_bazelrc "NO_WHOLE_ARCHIVE_OPTION" "1"
write_to_bazelrc "build --config=win-cuda"
write_to_bazelrc "test --config=win-cuda"
else
# If CUDA is enabled, always use GPU during build and test.
write_to_bazelrc "build --config=cuda"
write_to_bazelrc "test --config=cuda"
fi
# end of if "$TF_NEED_CUDA" == "1"