torch.accelerator.memory.get_memory_info

torch.accelerator.memory.get_memory_info(device_index=None, /) [источник]

Возвращает сведения о текущей памяти устройства с указанным индексом.

Параметры:

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

Возвращает:

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

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

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

tuple[int, int]