评价此页

torch.mtia.get_device_capability#

torch.mtia.get_device_capability(device=None)[source]#

返回给定设备的性能,以(主版本号,次版本号)的元组形式。

参数

device (torch.device or int, optional) – 如果 device 为 None(默认),则为当前设备的统计信息,由 current_device() 提供。

返回类型

tuple[int, int]