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