get_device_capability

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

获取设备计算能力。

参数

device (int|str|paddle.Place|None) - 设备、设备的 id 或设备的字符串名称，如 npu:x，从中获取设备的属性。如果输入为 None，则该设备为当前设备。

返回

tuple: (主版本号, 次版本号)

代码示例

          >>> 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)

         