About Scalene Scalene is a high-performance CPU, GPU and memory profiler for Python that does a number of things that other Python profilers do not and cannot do. It runs orders of magnitude faster ...

The pinned CPU tensors can't be released back to the OS immediately. import torch import gc import ctypes import psutil import os def get_memory_usage(): """Return current process RSS memory usage in ...