Skip to content

Commit a8576ce

Browse files
Update instructions for building the wheel from sources.
PiperOrigin-RevId: 748684104
1 parent 18c3a45 commit a8576ce

File tree

1 file changed

+13
-6
lines changed

1 file changed

+13
-6
lines changed

Diff for: site/en/install/source.md

+13-6
Original file line numberDiff line numberDiff line change
@@ -268,23 +268,23 @@ depending on the provided name, package will be created, e.g:
268268

269269
To build tensorflow CPU package:
270270
<pre class="devsite-terminal devsite-click-to-copy">
271-
bazel build //tensorflow/tools/pip_package:wheel --repo_env=WHEEL_NAME=tensorflow_cpu
271+
bazel build //tensorflow/tools/pip_package:wheel --repo_env=USE_PYWRAP_RULES=1 --repo_env=WHEEL_NAME=tensorflow_cpu
272272
</pre>
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 --config=cuda_wheel
276+
bazel build //tensorflow/tools/pip_package:wheel --repo_env=USE_PYWRAP_RULES=1 --repo_env=WHEEL_NAME=tensorflow --config=cuda --config=cuda_wheel
277277
</pre>
278278

279279
To build tensorflow TPU package:
280280
<pre class="devsite-terminal devsite-click-to-copy">
281-
bazel build //tensorflow/tools/pip_package:wheel --repo_env=WHEEL_NAME=tensorflow_tpu --config=tpu
281+
bazel build //tensorflow/tools/pip_package:wheel --repo_env=USE_PYWRAP_RULES=1 --repo_env=WHEEL_NAME=tensorflow_tpu --config=tpu
282282
</pre>
283283

284284
To build nightly package, set `tf_nightly` instead of `tensorflow`, e.g.
285285
to build CPU nightly package:
286286
<pre class="devsite-terminal devsite-click-to-copy">
287-
bazel build //tensorflow/tools/pip_package:wheel --repo_env=WHEEL_NAME=tf_nightly_cpu
287+
bazel build //tensorflow/tools/pip_package:wheel --repo_env=USE_PYWRAP_RULES=1 --repo_env=WHEEL_NAME=tf_nightly_cpu
288288
</pre>
289289

290290
As a result, generated wheel will be located in
@@ -354,7 +354,10 @@ virtual environment:
354354
<pre class="devsite-disable-click-to-copy prettyprint lang-bsh">
355355
<code class="devsite-terminal tfo-terminal-root">./configure # if necessary</code>
356356

357-
<code class="devsite-terminal tfo-terminal-root">bazel build //tensorflow/tools/pip_package:wheel --repo_env=WHEEL_NAME=tensorflow_cpu --config=opt</code>
357+
<code class="devsite-terminal tfo-terminal-root">
358+
bazel build //tensorflow/tools/pip_package:wheel \
359+
--repo_env=USE_PYWRAP_RULES=1 --repo_env=WHEEL_NAME=tensorflow_cpu --config=opt
360+
</code>
358361
`
359362
<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>
360363
</pre>
@@ -405,7 +408,11 @@ with GPU support:
405408
<pre class="devsite-disable-click-to-copy prettyprint lang-bsh">
406409
<code class="devsite-terminal tfo-terminal-root">./configure # if necessary</code>
407410

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>
411+
<code class="devsite-terminal tfo-terminal-root">
412+
bazel build //tensorflow/tools/pip_package:wheel \
413+
--repo_env=USE_PYWRAP_RULES=1 --repo_env=WHEEL_NAME=tensorflow --config=cuda \
414+
--config=cuda_wheel --config=opt
415+
</code>
409416

410417
<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>
411418
</pre>

0 commit comments

Comments
 (0)