torch.mtia.get_device_capability

torch.mtia.get_device_capability(device=None)

返回给定设备的能力，格式为 (主版本号, 次版本号) 的元组。

参数

device (torch.device 或 int, 可选) – 如果 device 为 None（默认值），则返回当前设备（通过 current_device() 获取）的能力统计信息。

返回类型

tuple[int, int]