torch.cuda.list_gpu_processes¶
- torch.cuda.list_gpu_processes(device=None)[source][source]¶
返回给定设备上正在运行的进程及其 GPU 内存使用情况的人类可读输出。
这在训练期间定期显示或处理内存不足异常时很有用。
- 参数
device (torch.device 或 int, 可选) – 选定的设备。如果
device为None(默认值)，则返回由current_device()指定的当前设备的输出。- 返回类型
