megengine.get_cuda_compute_capability

get_cuda_compute_capability(device, device_type=DeviceType.CUDA)[source]

Gets compute capability of the specified device.

Parameters

device (int) – device number.

Return type

int

Returns

a version number, or SM version.