/profiler-ui

The TensorFlow Profiler (TFProf) UI provides a visual interface for profiling TensorFlow models.

Primary LanguageHTMLApache License 2.0Apache-2.0

TensorFlow Profiler UI

The TensorFlow Profiler (TFProf) UI provides a visual interface for profiling TensorFlow models.

Installation

  1. Install Python dependencies. 
    pip install --user -r requirements.txt
  2. Install pprof.
  3. Create a profile context file using the tf.contrib.tfprof.ProfileContext class.
  4. Start the UI. 
    python ui.py --profile_context_path=/path/to/your/profile.context

Using a docker container

Building the image

docker build -t profiler .

Running in a container

docker run -it -p 7007:7007 --mount type=bind,source=PATH_TO_PROFILE_CONTEXT_FILE,target=/tmp/profile profiler --profile_context_path=/tmp/profile

Learn more

You can learn more about the TensorFlow Profiler's Python API and CLI here.

Screenshot

Browser support

Currently only Chrome is supported.

Contributing

Please see our contributor's guide

Feature requests

Want ideas for ways to contribute to the TensorFlow Profiler UI? Here are some requested features:

  • Support multiple profile contexts at once (#11)