torch.mtia.get_device_capability

torch.mtia.get_device_capability(device=None)

傳回指定設備的功能，格式為 (主要版本, 次要版本) 的元組。

參數

device ( torch.device 或 int, 選填) – 當 device 為 None 時（預設），由 current_device() 提供目前裝置的統計資訊。

回傳類型

Tuple[int, int]