get_device_capability

paddle.cuda. get_device_capability ( device: DeviceLike = None ) → tuple[int, int] [source]

Get the compute capability (major, minor) of a 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

A tuple (major, minor) representing the compute capability of the CUDA device.

Return type

tuple[int, int]

Examples

>>> 
>>> import paddle

# Get compute capability of the current CUDA device
>>> capability = paddle.cuda.get_device_capability()
>>> print(capability)  # e.g., (8, 0)

# Get compute capability of device cuda:0
>>> capability0 = paddle.cuda.get_device_capability("cuda:0")
>>> print(capability0)