memory_reserved

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

返回当前设备上由缓存分配器管理的内存总量。

参数

device (DeviceLike) - 指定要查询的设备，可以是 "int" 用来表示设备 id，可以是形如 "gpu:0" 之类的设备描述字符串，也可以是 paddle.CUDAPlace(0) 之类的设备实例。如果为 None（默认值）或未指定设备索引，则返回由 paddle.device.get_device() 给出的当前设备的统计信息。

返回

int, 当前设备上由缓存分配器管理的内存总量（字节）。

代码示例

          >>> import paddle
>>> # Get memory reserved for current device
>>> mem_reserved = paddle.cuda.memory_reserved()
>>> print(f"Memory reserved: {mem_reserved} bytes")
>>> # Get memory reserved for specific device
>>> mem_reserved = paddle.cuda.memory_reserved(0)
>>> print(f"Memory reserved on device 0: {mem_reserved} bytes")

         