cuda.bindings.nvml.device_get_cuda_compute_capability#

cuda.bindings.nvml.device_get_cuda_compute_capability(intptr_t device) → tuple#

Retrieves the CUDA compute capability of the device.

For all products. Returns the major and minor compute capability version numbers of the device. The major and minor versions are equivalent to the CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MINOR and CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR attributes that would be returned by CUDA’s cuDeviceGetAttribute().

Parameters:

device (intptr_t) – The identifier of the target device.

Returns:

int: Reference in which to return the major CUDA compute
capability.
int: Reference in which to return the minor CUDA compute
capability.

Return type:

A 2-tuple containing