torch.cuda.memory_summary¶
- torch.cuda.memory_summary(device=None, abbreviated=False)[来源][来源]¶
返回给定设备当前内存分配器统计信息的可读文本输出。
这在训练期间定期显示或处理内存不足异常时非常有用。
- 参数
device (torch.device or int, optional) – 选定的设备。如果 device 为
None（默认），则返回当前设备（由current_device()指定）的文本输出。abbreviated (bool, optional) – 是否返回缩写摘要（默认值：False）。
- 返回类型
注意
有关 GPU 内存管理的更多详细信息，请参阅内存管理。
