get_device_capability¶
返回从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))