Rate this Page
★★★★★

torch.cuda.memory.reset_peak_memory_stats#

torch.cuda.memory.reset_peak_memory_stats(device=None)[source]#

Reset the “peak” stats tracked by the CUDA memory allocator.

Seememory_stats() for details. Peak stats correspond to the“peak” key in each individual stat dict.

Parameters

device (torch.device orint,optional) – selected device. Returnsstatistic for the current device, given bycurrent_device(),ifdevice isNone (default).

Note

SeeMemory management for more details about GPU memorymanagement.