Hi, I am trying to calculate the peak memory utilization in pyych. While using CUDA, I can do torch.cuda.max_memory_allocated() to get the peak memory utilization. But there is no such option for CPU. For this reason, I am using “psutil.Process().memory_info().rss” to get the memory utilization. But this ends up giving me results that are wildly different from cuda memory usage. What can I do here ?