torch.accelerator.memory.max_memory_allocated

torch.accelerator.memory.max_memory_allocated(device_index=None, /) [source]

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

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

Параметры:

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

Возвращает:

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

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

int