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