19:T5ef,

nvtx Python package

Installation

Install using conda:

conda install -c conda-forge nvtx

...or using pip:

python -m pip install nvtx

Quick start

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()

Use NVIDIA Nsight Systems to
collect profile data:
```
nsys profile -t nvtx python demo.py
```
Visualize the results using the Nsight systems GUI:

Documentation

See here for detailed docs.

Release Notes

0.2.13

Fix tracing with tools that do not implement nvtxDomainIsEnabled.
Fix decorator ranges not ending when an exception is thrown.
Fix automatic function name annotation for decorator ranges.

0.2.12

Expose API via domain object for better performance.
No-OP when domain is disabled.
Improved documentation.

18:["$","div",null,{"className":"markdown-content max-w-2xl","dangerouslySetInnerHTML":{"__html":"$19"}}]

Dependency	Optional
Cython	true
nvidia_sphinx_theme	true
pytest	true
setuptools	true
sphinx	true

Package profile

nvtx

Releases

Release table

PyPI Downloads

Dependencies

Details

Project URLs

Dependencies

Package description