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.