Skip to content

Instantly share code, notes, and snippets.

View eqy's full-sized avatar
💭
damn that's crazy

eqy

💭
damn that's crazy
View GitHub Profile
@eqy
eqy / nsight.sh
Last active December 21, 2022 02:30 — forked from mcarilli/nsight.sh
Favorite nsight systems profiling commands for Pytorch scripts
# This isn't supposed to run as a bash script, i named it with ".sh" for syntax highlighting.
# https://developer.nvidia.com/nsight-systems
# https://docs.nvidia.com/nsight-systems/profiling/index.html
# My preferred nsys (command line executable used to create profiles) commands
#
# In your script, write
# torch.cuda.nvtx.range_push("region name")
# ...