PyTorchProfiler crashes when emit_nvtx=True #6153
Labels
bug
Something isn't working
callback
help wanted
Open to be worked on
priority: 0
High priority task
🐛 Bug
When training with PyTorchProfiler(emit_nvtx=True), the training stops with the following error :
AttributeError: 'emit_nvtx' object has no attribute 'function_events'
Please reproduce using the BoringModel
-> https://colab.research.google.com/drive/1cqMxMgDVgltluaYZAkn9d2srSDmOu-1p?usp=sharing
To Reproduce
Use following BoringModel and post here
Expected behavior
Environment
Additional context
Additional docs issue :
It says here (https://pytorch-lightning.readthedocs.io/en/1.2.0/advanced/profiler.html) that you have to use
nvprof
to collect the profiler traces, but it is no longer supported for devices with compute capability 8.0 and higher.I think that an exemple using nsight-compute would be great ( I can't test it right now because of the nvtx issue )
The text was updated successfully, but these errors were encountered: