torch.mps.driver_allocated_memory

torch.mps.driver_allocated_memory() [source]

Returns total GPU memory allocated by Metal driver for the process in bytes.

Note

The returned size includes cached allocations in MPSAllocator pools as well as allocations from MPS/MPSGraph frameworks.