torch.mps.current_allocated_memory

torch.mps.current_allocated_memory() [source]

Возвращает текущий объём памяти GPU в байтах, занятый тензорами.

Примечание

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

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

int