torch.cuda.memory.memory_summary

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

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

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

参数

• device (torch.device 或 int, optional) – 所选设备。如果 device 为 None (默认)，则返回当前设备 (current_device()) 的打印输出。

• abbreviated (bool, optional) – 是否返回缩略摘要 (默认: False)。

返回类型

str

注意

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