torch.mtia.get_device_capability

torch.mtia.get_device_capability(device=None) [исходный код]

Возвращает возможности указанного устройства в виде кортежа (основная версия, дополнительная версия).

Параметры:

device (torch.device или int, необязательный) – сведения о текущем устройстве, полученные с помощью current_device(), если device равен None (значение по умолчанию).

Тип возвращаемого значения:

tuple[int, int]