torch.accelerator.memory.memory_allocated

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

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

Параметры:

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

Возвращает:

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

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

int