評價此頁

torch.mtia.get_device_capability#

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

返回給定裝置的效能,格式為 (主版本號, 次版本號) 元組。

引數

device (torch.deviceint, optional) – 如果 device 為 None(預設值),則統計當前裝置(由 current_device() 給出)。

返回型別

tuple[int, int]