Install using conda:

conda install -c conda-forge nvtx

...or using pip:

python -m pip install nvtx

1. Annotate the parts of your Python code you wish to profile/analyze:

   # demo.py
   
   import time
   import nvtx
   
   @nvtx.annotate(color="blue")
   def my_function():
       for i in range(5):
           with nvtx.annotate("my_loop", color="red"):
               time.sleep(i)
   
   my_function()

2. Use NVIDIA Nsight Systems to collect profile data:

   nsys profile -t nvtx python demo.py
   

3. Visualize the results using the Nsight systems GUI:

   

See here for detailed docs.