torch.cuda.reset_max_memory_allocated¶
- torch.cuda.reset_max_memory_allocated(device=None)[源代码]¶
重置特定设备上张量所占用的最大 GPU 内存的跟踪起点。
有关详细信息,请参阅
max_memory_allocated()
。- 参数
device (torch.device 或 int, 可选) – 所选设备。如果
device
为None
(默认值),则返回由current_device()
给出的当前设备的统计信息。
警告
此函数现在调用
reset_peak_memory_stats()
,该函数会重置/所有/峰值内存统计信息。注意
有关 GPU 内存管理的更多详细信息,请参阅 内存管理。