Skip to main content

A simple tool for profiling and tracing PyTorch programs on NVIDIA GPUs

Project description

# DND: Dnd’s Not Dlprof!

[![PyPI version](https://badge.fury.io/py/dndlprof.svg)](https://badge.fury.io/py/dndlprof)

DND is a simple tool for profiling and tracing PyTorch programs on NVIDIA GPUs. If uses NVIDIA’s NSIGHT Systems and NSIGHT Compute along-side PyTorch 2.0’s Dynamo graph capturing system to provide framework operator and CUDA kernel level breakdowns of a Deep Learning application. Check out the [Sample App](./dnd/sample_app.py) for an example of an instrumented application.

## Install `bash $ pip install dndlprof `

## Quick Start `bash $ dnd -- python -m dnd.sample_app `

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

dndlprof-0.9.0.tar.gz (14.0 kB view hashes)

Uploaded Source

Built Distribution

dndlprof-0.9.0-py3-none-any.whl (16.4 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page