torch.cuda.get_device_name

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

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

Параметры:

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

Возвращает:

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

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

str