cuda.bindings.nvml.device_get_num_gpu_cores#

cuda.bindings.nvml.device_get_num_gpu_cores(intptr_t device) → unsigned int#

Gets the device’s core count.

Parameters:: device (intptr_t) – The identifier of the target device.
Returns:: The number of cores for the specified device.
Return type:: unsigned int

Note

On MIG-enabled GPUs, querying the device’s core count is currently not supported using this API. Please use nvmlDeviceGetGpuInstanceProfileInfo to fetch the MIG device’s core count.