torch.cuda.list_gpu_processes¶
- torch.cuda.list_gpu_processes(device=None)[原始碼][原始碼]¶
傳回指定裝置上正在執行的處理程序及其 GPU 記憶體使用情況的人工可讀列印輸出。
在訓練期間定期顯示，或在處理記憶體不足的例外情況時，這可能很有用。
- 參數
device ( torch.device 或 int, 選填) – 選定的裝置。如果
device是None(預設值)，則會印出目前裝置的資訊，該裝置由current_device()提供。- 回傳類型
