torch.cuda.列出GPU进程

torch.cuda.list_gpu_processes(device=None)

返回给定设备的运行进程及其GPU内存使用情况的可读打印输出。

这在训练期间定期显示或处理内存不足异常时非常有用。

Parameters

设备 (torch.device 或 整数, 可选) – 选择的设备。返回当前设备的打印信息，由 current_device() 提供，如果 device 为 None（默认）。

Return type

str