评价此页

torch.mtia.get_device_capability#

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

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

参数

device (torch.deviceint, optional) – 如果 device 为 None(默认值),则统计当前设备(由 current_device() 给出)。

返回类型

tuple[int, int]