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