torch.cuda.get_device_name

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

Получает имя устройства.

Параметры

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

Возвращает

имя устройства

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

str