The TensorFlow Profiler (TFProf) UI provides a visual interface for profiling TensorFlow models.
- Install Python dependencies.
pip install --user -r requirements.txt
- Install pprof.
- Create a profile context file using the tf.contrib.tfprof.ProfileContext class.
- Start the UI.
python ui.py --profile_context_path=/path/to/your/profile.context
You can learn more about the TensorFlow Profiler's Python API and CLI here.
Currently only Chrome is supported.
Please see our contributor's guide