pytorch_profiler2880×1210 813 KB

Since I am not the admin of the remote server, I couldn’t directly run nvprof. I actually ran
with torch.autograd.profiler.profile(use_cuda=True) as prof:
     model(imgs)

Something like this
I am not able to create profile for Tensorflow code as of now. It is not easy as Pytorch.
But is it okay with just this ?