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