torch.cuda.list_gpu_processes¶
- torch.cuda.list_gpu_processes(device=None)[source][source]¶
返回給定裝置上正在執行的程序及其 GPU 記憶體使用情況的人類可讀輸出。
這在訓練期間定期顯示或處理記憶體不足異常時很有用。
- 引數
device (torch.device 或 int, 可選) – 選定的裝置。如果
device為None(預設值)，則返回由current_device()指定的當前裝置的輸出。- 返回型別
