Rate this Page

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.

Return type

int