torch.cuda.memory_summary

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

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

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

参数:

• 设备（torch.device 或 int，可选）- 选择设备。如果 device 是 None （默认），则返回当前设备 current_device() 的打印输出。

• 简略（bool，可选）- 是否返回简略摘要（默认：False）。

返回类型:

str

注意

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