memory_allocated¶

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

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

注解

Paddle中分配给Tensor的显存块大小会进行256字节对齐，因此可能大于Tensor实际需要的显存大小。例如，一个shape为[1]的float32类型Tensor会占用256字节的显存，即使存储一个floatt32类型数据实际只需要4字节。

参数¶

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

返回¶

一个整数，表示给定设备上当前分配给Tensor的显存大小，以字节为单位。

代码示例¶

          # required: gpu
import paddle

memory_allocated_size = paddle.device.cuda.memory_allocated(paddle.CUDAPlace(0))
memory_allocated_size = paddle.device.cuda.memory_allocated(0)
memory_allocated_size = paddle.device.cuda.memory_allocated("gpu:0")

         