get_device_capability¶

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

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

参数¶

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

返回¶

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

代码示例¶

          >>> import paddle
>>> paddle.device.set_device('gpu')
>>> paddle.device.cuda.get_device_capability()

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

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