megengine.get_cuda_compute_capability

get_cuda_compute_capability(device, device_type=DeviceType.CUDA)

获取指定设备的计算能力。

参数:

device (int) – 设备序号。

返回类型:

int

返回:

一个版本号，或”CUDA SM的版本”