torch.cuda.memory_summary¶
- torch.cuda.memory_summary(device=None, abbreviated=False)[來源][來源]¶
傳回指定裝置目前記憶體配置器統計資料的人工可讀列印輸出。
在訓練期間定期顯示，或在處理記憶體不足例外狀況時，這會很有用。
- 參數
device ( torch.device 或 int, 可選) – 選定的裝置。如果
device為None(預設)，則會回傳由current_device()給定的目前裝置的列印輸出。abbreviated ( bool, 可選) – 是否回傳縮寫的摘要 (預設：False)。
- 回傳類型
注意
有關 GPU 記憶體管理的更多詳細資訊，請參閱記憶體管理。
