torch.cuda.memory.memory_summary

torch.cuda.memory.memory_summary(device=None, abbreviated=False)

返回给定设备的当前内存分配器统计信息的易读打印输出。

这在训练期间或处理内存不足异常时，可以作为定期显示的有用工具。

参数

• device (torch.device 或 int, 可选) – 选定的设备。如果 device 为 None（默认值），则返回当前设备（由 current_device() 指定）的打印输出。

• abbreviated (bool, 可选) – 是否返回缩略摘要（默认值：False）。

返回类型

str

注意

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