torch.cuda.memory_snapshot

torch.cuda.memory_snapshot()

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

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

注意

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