torch.cuda.memory_summary¶
- torch.cuda.memory_summary(device=None, abbreviated=False)[來源][來源]¶
返回給定裝置當前記憶體分配器統計資訊的可讀文字輸出。
這在訓練期間定期顯示或處理記憶體不足異常時非常有用。
- 引數
device (torch.device or int, optional) – 選定的裝置。如果 device 為
None（預設），則返回當前裝置（由current_device()指定）的文字輸出。abbreviated (bool, optional) – 是否返回縮寫摘要（預設值：False）。
- 返回型別
注意
有關 GPU 記憶體管理的更多詳細資訊，請參閱記憶體管理。
