torch.xpu.memory.list_gpu_processes

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

Возвращает список выполняющихся процессов и использования ими памяти графического процессора на указанном устройстве.

Параметры:

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

Возвращает:

Удобная для чтения сводка по каждому выполняющемуся процессу и объёму используемой им памяти графического процессора в МБ.

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

str

Примечание

Состояние процессов приводится на уровне физического устройства и включает все процессы, связанные с этим устройством.