torch.xpu.memory.memory_reserved

torch.xpu.memory.memory_reserved(device=None) [исходный код]

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

Параметры:

device (torch.device или int или str, необязательный) – выбранное устройство. Возвращает статистику для текущего устройства, заданного функцией current_device(), если device имеет значение None (по умолчанию).

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

int