torch.cuda.memory_snapshot

torch.cuda.memory_snapshot()[源代码]

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

要解释该函数的输出，需要了解内存分配器的内部工作原理。

注意

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