torch.cuda.memory_snapshot¶ torch.cuda.memory_snapshot()[源代码]¶ 返回所有设备上 CUDA 内存分配器状态的快照。 解释此函数的输出需要熟悉内存分配器内部机制。 注意 有关 GPU 内存管理的更多详细信息,请参阅 内存管理。