torch.xpu.memory.mem_get_info

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

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

Параметры:

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

Возвращает:

кортеж из двух целых чисел (free_memory, total_memory) в байтах.

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

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

tuple[int, int]