reset_max_memory_allocated¶

paddle.device.cuda. reset_max_memory_allocated ( device: _CudaPlaceLike | None = None ) → None [source]

Reset the peak size of GPU memory that is allocated to tensor of the given device.

Parameters: device (paddle.CUDAPlace|int|str|None, optional) – The device, the id of the device or the string name of device like ‘gpu:x’. If device is None, the device is the current device. Default: None.

Examples

           >>> 
>>> 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")

          