torch.accelerator.memory.memory_reserved

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

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

Параметры:

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

Возвращает:

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

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

int