torch.xpu.device_memory_used

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

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

Параметры:

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

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

int

Примечание

Для доступа к сведениям об использовании памяти GPU этому API могут потребоваться повышенные привилегии (например, sudo).