torch.cuda.memory_snapshot

torch.cuda.memory_snapshot()[source][source]

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

理解此函数的输出需要熟悉内存分配器内部机制。

注意

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