Problem to load the torch.profiler module

I am studying the tutorial given here to profile the CPU time of the model. I get an error message ModuleNotFoundError: No module named 'torch.profiler'. I have torch version 1.7.1 installed. May I know what’s the problem?

The new profiler was released in 1.8.1 as described in this blog post so you would have to update PyTorch.

@ptrblck It worked. Thanks!