memory_allocated

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

返回给定设备上当前分配给 Tensor 的显存大小。

参数

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

返回

int, 当前设备上分配的内存总量（字节）。

代码示例

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

         