[go: up one dir, main page]

Skip to content

Base code for profiling DNN models on Nvidia Hardware

Notifications You must be signed in to change notification settings

MohanadOdema/DLProf

Repository files navigation

Trial code for DLprof on Nvidia GPUs and hardware

Installation

python3 -m pip install nvidia-pyindex python3 -m pip install nvidia-dlprof

Run for pytorch dummy network

dlprof --mode=pytorch --force true python3 dummy_network.py dlprof --mode=pytorch --database=nsys_profile.sqlite

Reports Generation

dlprof --mode=pytorch --reports=all --database=nsys_profile.sqlite

Reference

https://docs.nvidia.com/deeplearning/frameworks/dlprof-user-guide/

About

Base code for profiling DNN models on Nvidia Hardware

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages