max_memory_reserved¶

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

返回给定设备上由 Allocator 管理的显存峰值。

参数¶

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

返回¶

一个整数，表示给定设备上当前由 Allocator 管理的显存峰值，以字节为单位。

代码示例¶

          # required: gpu
import paddle

max_memory_reserved_size = paddle.device.cuda.max_memory_reserved(paddle.CUDAPlace(0))
max_memory_reserved_size = paddle.device.cuda.max_memory_reserved(0)
max_memory_reserved_size = paddle.device.cuda.max_memory_reserved("gpu:0")

         