torch.cuda.memory.memory_snapshot

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

返回所有设备上 CUDA 内存分配器状态的快照。

要解释此函数返回的输出，需要熟悉内存分配器内部。

注意

有关 GPU 内存管理的更多详细信息，请参阅 内存管理。