Как узнать информацию о видеокарте в python?

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

import pycuda.driver as cuda

# Инициализация драйвера CUDA
cuda.init()

# Получение числа доступных устройств CUDA
device_count = cuda.Device.count()

# Перебор устройств CUDA и получение информации о них
for i in range(device_count):
    device = cuda.Device(i)
    
    # Название устройства
    print("Устройство {}: {}".format(i, device.name()))
    
    # Число мультипроцессоров
    print("Мультипроцессоры: {}".format(device.multiprocessor_count()))
    
    # Объем глобальной памяти
    print("Объем глобальной памяти: {} MB".format(device.total_memory() // (1024**2)))
    
    # Максимальное число нитей на блок
    print("Максимальное число нитей на блок: {}".format(device.max_threads_per_block()))
    
    # Максимальный размер блока
    print("Максимальный размер блока: {}".format(device.max_block_dim()))
    
    # Максимальный размер сетки
    print("Максимальный размер сетки: {}".format(device.max_grid_dim()))

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18

import GPUtil

# Получение информации о доступных видеокартах
gpus = GPUtil.getGPUs()

# Перебор видеокарт и получение информации о них
for gpu in gpus:
    # Название видеокарты
    print("Видеокарта: {}".format(gpu.name))
    
    # Использование видеокарты (в процентах)
    print("Использование: {}%".format(gpu.load * 100))
    
    # Температура видеокарты (в градусах Цельсия)
    print("Температура: {}°C".format(gpu.temperature))
    
    # Объем памяти видеокарты (в MB)
    print("Объем памяти: {} MB".format(gpu.memoryTotal))

1

pip install pycuda

1

pip install gputil