torch.cuda.list_gpu_processes

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

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

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

Параметры

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

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

str