torch.cuda.memory.list_gpu_processes

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

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

Это может быть полезно для периодического отображения во время обучения или при обработке исключений нехватки памяти.

Параметры:

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

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

str