reset_peak_memory_stats

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

重置所有设备的峰值内存统计信息。此方法重置程序执行期间为每个设备记录的峰值内存使用情况。它将所有设备的峰值内存使用率设置为零。与 reset_peak_memory_stats 功能一致

参数

device (int|paddle.Place|str|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")

         