get_device_capability

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

获取设备计算能力。

参数

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

返回

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

代码示例

>>> import paddle
>>> cap = paddle.device.get_device_capability()
>>> print(cap)