torch.mps.driver_allocated_memory
-
torch.mps.driver_allocated_memory()[source] -
Возвращает общее количество памяти видеокарты, выделенной драйвером Metal для процесса в байтах.
Примечание
Возвращаемый размер включает кэшированные выделения в пулах MPSAllocator, а также выделения из фреймворков MPS/MPSGraph.
- Тип возвращаемого значения
© 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
