get_device_capability

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

返回从CUDA函数 cudaDeviceProp 获取到的定义设备计算能力的主要和次要修订号。

参数

  • device (paddle.CUDAPlace|int, 可选) - 希望获取计算能力的设备或者设备ID。如果device为None(默认),则为当前的设备。

返回

tuple(int,int): 设备计算能力的主要和次要修订号。

代码示例

# required: gpu

import paddle

paddle.device.cuda.get_device_capability()

paddle.device.cuda.get_device_capability(0)

paddle.device.cuda.get_device_capability(paddle.CUDAPlace(0))