reset_peak_memory_stats

paddle.device. reset_peak_memory_stats ( device=None ) [源代码] 

重置设备的峰值内存统计信息。

参数

device (int|paddle.CUDAPlace|None) - 设备、设备的 id 或设备的字符串名称，如 npu:x'，从中获取设备的属性。如果设备为 None，则该设备为当前设备，默认值：None。

代码示例

          >>> import paddle
>>> paddle.device.set_device('gpu')  # or '<custom_device>'

>>> # paddle.cuda.reset_max_memory_allocated() is equivalent to paddle.device.reset_max_memory_allocated()

>>> paddle.device.reset_max_memory_allocated(paddle.CUDAPlace(0))
>>> paddle.device.reset_max_memory_allocated(0)
>>> paddle.device.reset_max_memory_allocated("gpu:0")

         