torch.cuda.memory_summary

torch.cuda.memory_summary(device=None, abbreviated=False)[源代码]

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

这在训练过程中定期显示，或是处理内存不足异常时会非常有用。

参数

• device (torch.device 或 int, 可选) – 选定的设备。如果 device 为 None（默认值），则返回当前设备的信息，该信息由current_device() 提供。

• 简略模式 (bool, 可选) – 是否以简略形式返回摘要（默认值：False）。

返回类型

str

注意

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