torch.cuda.list_gpu_processes

torch.cuda.list_gpu_processes(device=None) [source]

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

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

Параметры:

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

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

str