get_device_capability¶
- paddle.device.cuda. get_device_capability ( device=None ) ¶
返回从 CUDA 函数 cudaDeviceProp 获取到的定义设备计算能力的主要和次要修订号。
参数¶
device (paddle.CUDAPlace|int,可选) - 希望获取计算能力的设备或者设备 ID。如果 device 为 None(默认),则为当前的设备。
返回¶
tuple(int,int):设备计算能力的主要和次要修订号。
代码示例¶
COPY-FROM: paddle.device.cuda.get_device_capability