Функция get_device_name

Функция get_device_name возвращает строку с именем устройства CUDA по его индексу. Первым параметром функция принимает целое число - индекс устройства. Если индекс не указан, используется текущее устройство по умолчанию.

Синтаксис

torch.cuda.get_device_name([device_index])

Пример

Давайте получим имя текущего устройства CUDA:

import torch

if torch.cuda.is_available():
	name = torch.cuda.get_device_name()
	print(name)

Результат выполнения кода:

"NVIDIA GeForce RTX 3080"

Пример

Давайте получим имя устройства по конкретному индексу:

import torch

if torch.cuda.is_available():
	device_count = torch.cuda.device_count()
	for i in range(device_count):
		name = torch.cuda.get_device_name(i)
		print(f"Device {i}: {name}")

Результат выполнения кода для системы с двумя GPU:

"Device 0: NVIDIA GeForce RTX 3080"
"Device 1: NVIDIA GeForce RTX 3060"

Пример

Давайте проверим, что устройство с заданным индексом существует, перед получением имени:

import torch

if torch.cuda.is_available():
	index = 0
	if index < torch.cuda.device_count():
		name = torch.cuda.get_device_name(index)
		print(f"Device name: {name}")
	else:
		print("Device not found")

Результат выполнения кода:

"Device name: NVIDIA GeForce RTX 3080"

Смотрите также

функцию is_available,
которая проверяет доступность CUDA
функцию device_count,
которая возвращает количество устройств CUDA
функцию current_device,
которая возвращает индекс текущего устройства
функцию get_device_properties,
которая возвращает свойства устройства CUDA