torch.cuda.get_device_capability

torch.cuda.get_device_capability(device=None) [source]

Получает характеристики CUDA устройства.

Параметры:

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

Возвращает:

главную и второстепенную характеристики CUDA устройства

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

tuple(int, int)