torch.mtia.get_device_capability

torch.mtia.get_device_capability(device=None)

返回指定设备的性能，以（主版本号，副版本号）元组的形式。

参数:

device（torch.device 或 int，可选）- 当前设备统计信息，由 current_device()提供，如果 device 为 None（默认）。

返回类型:

元组[int, int]