It wasn't obvious on PyTorch's documentation of how to use PyTorch Profiler (as of today, 8/12/2021), so I have spent some time to understand how to use it and this gist contains a simple example to use.
- Install the required packages:
python>=1.9.0
torchvision>=0.10.0
numpy
matplotlib