get_device_properties

paddle.cuda. get_device_properties ( device: DeviceLike = None ) [source]

Get the properties of a CUDA device.

Parameters

device (int | str | paddle.CUDAPlace | paddle.CustomPlace | None, optional) –

The target device to query.

• None: use the current device.

• int: device index (e.g., 0 -> ‘gpu:0’).

• str: device string (e.g., “cuda:0”, “gpu:1”).

• CUDAPlace or CustomPlace: Paddle device objects.

Returns

An object containing the device properties, such as name, total memory, compute capability, and multiprocessor count.

Return type

DeviceProperties

Examples

>>> 
>>> import paddle

# Get the properties of the current device
>>> props = paddle.cuda.get_device_properties()
>>> print(props)