tensorplay.cuda.reset_peak_memory_stats

tensorplay.cuda.reset_peak_memory_stats(device: Any = None) → None[source]

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

Peak stats correspond to the “peak” key in each individual stat dict.

Parameters:

device (tensorplay.Device or int, optional) – selected device. Returns statistic for the current device, given by current_device(), if device is None (default).