Функция get_device_properties

Функция get_device_properties возвращает объект с подробной информацией о CUDA-устройстве. Первым параметром функция принимает индекс устройства (целое число). Если устройство не указано, используется устройство с индексом 0.

Синтаксис

torch.cuda.get_device_properties(device)

Параметры

Функция принимает следующие параметры:

device (int) - индекс CUDA-устройства, свойства которого нужно получить.

Возвращаемое значение

Функция возвращает объект типа torch._C._CudaDeviceProperties, содержащий следующие атрибуты:

name - имя устройства (например, "Tesla T4");
total_memory - общий объём памяти в байтах;
major и minor - основная и дополнительная версии вычислительных возможностей CUDA;
multi_processor_count - количество мультипроцессоров;
max_threads_per_multi_processor - максимальное число потоков на мультипроцессор;
is_multi_gpu_board - флаг многогранной конфигурации.

Пример

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

import torch

props = torch.cuda.get_device_properties(0)
print(props)

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

_CudaDeviceProperties(name='Tesla T4', major=7, minor=5, total_memory=15083651072, multi_processor_count=40)

Пример

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

import torch

props = torch.cuda.get_device_properties(0)
print(props.name)

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

"Tesla T4"

Пример

Давайте получим объём памяти и вычислительные возможности для устройства:

import torch

props = torch.cuda.get_device_properties(0)
memory_gb = props.total_memory / (1024 ** 3)
print(f"Memory: {memory_gb:.2f} GB")
print(f"Compute capability: {props.major}.{props.minor}")

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

"Memory: 14.05 GB"
"Compute capability: 7.5"

Пример

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

import torch

if torch.cuda.is_available():
	props = torch.cuda.get_device_properties(0)
	print(f"Device: {props.name}")
	print(f"Multiprocessors: {props.multi_processor_count}")
else:
	print("CUDA is not available")

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

"Device: Tesla T4"
"Multiprocessors: 40"

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

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