torch.cuda.memory_summary

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

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

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

参数

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

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

返回类型

str

注意

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