torch.mps.driver_allocated_memory

torch.mps.driver_allocated_memory() [исходный код]

Возвращает общий объём памяти GPU, выделенной драйвером Metal для процесса, в байтах.

Примечание

Возвращаемый размер включает кэшированные выделения в пулах MPSAllocator, а также выделения из фреймворков MPS/MPSGraph.

Тип возвращаемого значения:

int