get_device_capability

dragon.cuda.get_device_capability(device_id=None)[source]

Return the capability of specified device.

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

Parameters:
  • device_id (int, optional) – The device index.
Returns:

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