torch.accelerator.memory.max_memory_reserved

torch.accelerator.memory.max_memory_reserved(device_index=None, /) [исходный код]

Возвращает текущий максимальный объём памяти устройства ускорителя, управляемый кэширующим аллокатором, в байтах для указанного индекса устройства.

По умолчанию возвращается пиковый объём кэшированной памяти с начала работы этой программы. Для сброса начальной точки отслеживания этой метрики можно использовать reset_peak_memory_stats().

Параметры:

device_index (torch.device, str, int, необязательный) – индекс целевого устройства. Если не указан, по умолчанию используется torch.accelerator.current_device_index(). Если указано значение типа torch.device или str, его тип должен совпадать с типом текущего устройства ускорителя.

Возвращает:

пиковый объём памяти, зарезервированной PyTorch (в байтах) в текущем процессе.

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

int