Profiling on CPU

[rkruegs123]

Hi, I am trying to use the Perfetto profiling per this link: https://jax.readthedocs.io/en/latest/profiling.html

However, I get the following error: 2022-09-02 19:42:42.384729: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory

Is this profiling only intended for use on GPUs? If so, how can I profile on CPU?

[rkruegs123]

Resolved! If anybody runs into this, it is because there are tensorflow dependencies, so you have to uninstall tensorflow and install tensorflow-cpu instead.