Memory footprint profiling

Developer Tools CUPTI – CUDA Profiler Tools Interface
1

I wanna get memory footprint profilings and breakdowns when making inference using Pytorch Model. How can I do that for further analysis