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