torch.cuda.list_gpu_processes
- torch.cuda.list_gpu_processes(device=None)[源代码]
-
返回给定设备上正在运行的进程及其 GPU 内存使用的易读输出。
这在训练过程中定期显示，或是处理内存不足异常时会非常有用。
- 参数
-
device (torch.device 或 int, 可选) – 选定的设备。如果
device为None（默认值），则返回当前设备的信息，该信息由current_device()提供。 - 返回类型