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