torch.cuda.memory.memory_snapshot

torch.cuda.memory.memory_snapshot(mempool_id=None)

返回所有裝置上 CUDA 記憶體分配器狀態的快照。

要解釋此函式返回的輸出，需要熟悉記憶體分配器內部。

注意

有關 GPU 記憶體管理的更多詳細資訊，請參閱 記憶體管理。