mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-14 10:56:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

tbp nightly instructions

Browse Source
This commit is contained in:
Robert Dyro 2024-12-23 08:32:54 +01:00
parent 4a6cfebcea
commit 213e1782ac
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
docs/profiling.md
View File

@ -93,6 +93,12 @@ plugins" error described {ref}`below <multiple_installs>`. See
<https://www.tensorflow.org/guide/profiler> for more information on installing
TensorBoard.
Nightly version of TensorBoard profiler requires nightly tensorflow and
tensorboard
```shell
pip install tf-nightly tb-nightly tbp-nightly
```
### Programmatic capture
You can instrument your code to capture a profiler trace via the