torch.mps.driver_allocated_memory

torch.mps.driver_allocated_memory()

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.

Return type

int