Hi!
I have run into some CUPTI warning in PyTorch 1.9.0
In PyTorch 1.8.0,
with torch.autograd.profiler.profile(use_cuda=True) as prof:
 y = model(x)
prof.export_chrome_trace("trace.json")

The following code works and chrome trace shows both CPU and CUDA traces.
Whereas in PyTorch 1.9.0,
with torch.profiler.profile(activities=[torch.profiler.ProfilerActivity.CPU, torch.profiler.ProfilerActivity.CUDA]) as prof:
 y = model(x)
prof.export_chrome_trace("trace.json")

gives me a warning of /home/surya/.local/lib/python3.6/site-packages/torch/autograd/profiler.py:440: UserWarning: CUPTI tracing is not available, falling back to legacy CUDA profiling   warn("CUPTI tracing is not available, falling back to legacy CUDA profiling")  and shows only CPU traces.
Any idea why this warning shows up and how to fix it? Thanks in advance for your reply.
I have been working on with Jetson Xavier AGX and installed PyTorch binaries from here