torch.mtia.get_device_capability#
- torch.mtia.get_device_capability(device=None)[source]#
Return capability of a given device as a tuple of (major version, minor version).
- Parameters
device (torch.device or int, optional) – statistics for the current device, given by current_device(), if device is None (default).
- Return type