get_device_capability¶

dragon.cuda.get_device_capability(device_index=None)[source]¶

Return the capability of specified device.

If device_index is None, the current device will be selected.

Parameters:

• device_index (int, optional) – The device index.

Returns:

Tuple[int, int] – The major and minor number.