diff options
| -rw-r--r-- | tensorflow/g3doc/get_started/os_setup.md | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/g3doc/get_started/os_setup.md b/tensorflow/g3doc/get_started/os_setup.md index b087a8dbc3..345ed01387 100644 --- a/tensorflow/g3doc/get_started/os_setup.md +++ b/tensorflow/g3doc/get_started/os_setup.md @@ -858,9 +858,12 @@ default and if you want to limit RAM usage you can add `--local_resources ```bash $ bazel build -c opt //tensorflow/tools/pip_package:build_pip_package -# To build with GPU support: +# To build with support for CUDA: $ bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package +# Alternatively, to build with support for OpenCL: +$ bazel build -c opt --config=sycl //tensorflow/tools/pip_package:build_pip_package + $ bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg # The name of the .whl file will depend on your platform. |