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)