PyTorch 1.8.1 comes with PyTorch Profiler

PyTorch 1.8.1 introduces the PyTorch Profiler, which is a tool for performance analysis and troubleshooting for large-scale deep learning models.

Link to Full Article: Read Here