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
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.mps.driver_allocated_memory.html