torch.cuda.list_gpu_processes

torch.cuda.list_gpu_processes(device=None)

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

这可以在训练期间定期显示，或者在处理内存不足异常时使用。

参数:

设备（torch.device 或 int，可选）- 选择设备。如果 device 是 None （默认），则返回当前设备 current_device() 的打印输出。

返回类型:

str