Skip to content

Commit bc7239f

Browse files
Refactor hermetic CUDA flags and update --config=cuda to add CUDA dependencies both for bazel build and bazel test phases.
Add `--@local_config_cuda//cuda:override_include_cuda_libs` to override settings for TF wheel. Forbid building TF wheel with `--@local_config_cuda//cuda:include_cuda_libs=true` PiperOrigin-RevId: 666848518
1 parent d1ab9f4 commit bc7239f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

site/en/install/source.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -273,7 +273,7 @@ bazel build //tensorflow/tools/pip_package:wheel --repo_env=WHEEL_NAME=tensorflo
273273

274274
To build tensorflow GPU package:
275275
<pre class="devsite-terminal devsite-click-to-copy">
276-
bazel build //tensorflow/tools/pip_package:wheel --repo_env=WHEEL_NAME=tensorflow --config=cuda
276+
bazel build //tensorflow/tools/pip_package:wheel --repo_env=WHEEL_NAME=tensorflow --config=cuda --config=cuda_wheel
277277
</pre>
278278

279279
To build tensorflow TPU package:
@@ -405,7 +405,7 @@ with GPU support:
405405
<pre class="devsite-disable-click-to-copy prettyprint lang-bsh">
406406
<code class="devsite-terminal tfo-terminal-root">./configure # if necessary</code>
407407

408-
<code class="devsite-terminal tfo-terminal-root">bazel build //tensorflow/tools/pip_package:wheel --repo_env=WHEEL_NAME=tensorflow --config=cuda --config=opt</code>
408+
<code class="devsite-terminal tfo-terminal-root">bazel build //tensorflow/tools/pip_package:wheel --repo_env=WHEEL_NAME=tensorflow --config=cuda --config=cuda_wheel --config=opt</code>
409409

410410
<code class="devsite-terminal tfo-terminal-root">chown $HOST_PERMS bazel-bin/tensorflow/tools/pip_package/wheel_house/tensorflow-<var>version</var>-<var>tags</var>.whl</code>
411411
</pre>

0 commit comments

Comments
 (0)