torch.cuda.get_device_capability

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

Получить вычислительные возможности CUDA устройства.

Параметры:

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

Возвращает:

основную и дополнительную версии вычислительных возможностей CUDA устройства

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

tuple(int, int)