Profile Python function

In this example, we show how to profile any Python function.

We recommend running this example in NVIDIA NGC PyTorch container. The Python script profile.py runs profiling of provided Python function.

To run the profiling, run the script:

./profile.py