Skip to content

Release 3.1.0 corresponding to NGC container 20.06
Compare
Choose a tag to compare
@dzier dzier released this 29 Jun 23:18
v3.1.0
a1bcf67
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

PyProf - PyTorch Profiler for GPUs

PyProf is a tool that profiles and analyzes the GPU performance of PyTorch models. PyProf aggregates kernel performance from Nsight Systems or NvProf. The tool can identify the layer that launched the kernel, identify the tensor dimensions and precision, perform forward-backward correlation, determine Tensor Core usage, and correlate the line in the user's code that launched a particular kernel.

What's New In 3.1.0

Known Issues

  • This software only supports PyTorch 1.6.
Assets 2
Loading