reset_max_memory_allocated¶

paddle.device.cuda. reset_max_memory_allocated ( device=None ) [源代码] ¶

重置给定设备上分配给 Tensor 的显存峰值统计。

参数¶

device (paddle.CUDAPlace|int|str，可选) - 设备、设备 ID 或形如 gpu:x 的设备名称。如果 device 为 None，则 device 为当前的设备。默认值为 None。

返回¶

None

代码示例¶

          >>> import paddle
>>> paddle.device.set_device('gpu')

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

         