Как проверить статус и версию CUDA на видеокарте NVIDIA

Технология CUDA (Compute Unified Device Architecture) является фундаментальным инструментом для ускорения вычислений на графических процессорах NVIDIA. Она позволяет использовать мощь видеокарты не только для рендеринга графики, но и для задач машинного обучения, обработки видео и научных расчетов. Однако наличие видеокарты бренда NVIDIA не гарантирует автоматически, что ваш драйвер и операционная система корректно поддерживают требуемую версию архитектуры.

Прежде чем запускать ресурсоемкое приложение или нейросеть, необходимо убедиться, что ваш GPU способен выполнять параллельные вычисления через данный интерфейс. Проверка может быть проведена как для встроенных в систему утилит, так и через стороннее программное обеспечение. Это критически важно для предотвращения ошибок компиляции и сбоев в работе специализированного софта.

Процесс диагностики включает в себя несколько этапов: от визуального осмотра модели карты до запуска командной строки для получения детальной технической информации. Мы разберем каждый метод, чтобы вы могли получить исчерпывающие данные о состоянии вашего оборудования и драйверов.

Понимание термина и значение версии CUDA

Многие пользователи путают версию драйверов с версией CUDA Toolkit. Важно понимать, что драйвер видеокарты содержит встроенную поддержку определенной версии CUDA. Если вы установите драйвер, поддерживающий версию 11.8, это не означает, что вы можете запустить приложения, требующие версии 12.0, без обновления.

Каждое поколение видеокарт имеет свою архитектуру, обозначенную кодовыми именами (например, Ampere, Ada Lovelace, Volta). Архитектура определяет максимальную версию параллельных вычислений, которую железо способно поддерживать физически. Программное обеспечение лишь активирует этот потенциал при условии наличия соответствующих драйверов.

Для большинства современных задач, таких как обучение нейросетей или монтаж 4K-видео, рекомендуется иметь актуальную версию CUDA. Старые карты могут поддерживать только устаревшие версии, что ограничивает их применение в новых проектах. Проверка совместимости начинается с идентификации вашей модели.

Проверка через командную строку (nvidia-smi)

Самый быстрый и надежный способ узнать текущую версию — использовать встроенный инструмент nvidia-smi. Эта утилита поставляется с драйверами NVIDIA и не требует установки дополнительного ПО. Она выводит информацию о загрузке, температуре и, что важно для нас, о версии API.

Откройте командную строку или терминал. Введите следующую команду и нажмите Enter:

nvidia-smi

В открывшемся окне обратите внимание на правый верхний угол. Там будет указано значение CUDA Version. Это максимальная версия CUDA, которую поддерживает установленный драйвер. Если вы видите здесь 12.1, значит, вы можете запускать приложения, требующие версии до 12.1 включительно.

Также в таблице вы увидите название вашей видеокарты, например, GeForce RTX 3080. Для более детальной информации о выделенных ядрах и памяти можно использовать ключ -q:

nvidia-smi -q

⚠️ Внимание: Значение CUDA Version в выводе nvidia-smi показывает максимальную поддерживаемую версию драйвером, а не версию Toolkit, установленную в системе. Для разработки вам может потребоваться отдельная установка CUDA Toolkit.

Использование специализированных утилит

Если командная строка кажется вам неудобной, существуют графические инструменты, которые предоставляют более наглядную информацию. Одной из самых популярных является GPU-Z. Эта утилита легка и предоставляет детальные данные о каждой части видеокарты.

После запуска GPU-Z перейдите на вкладку Advanced и выберите пункт CUDA. Здесь вы увидите количество потоковых процессоров и версию поддержки. Если значок CUDA активен и отображает данные, значит, интерфейс работает корректно. Отсутствие данных может указывать на проблемы с драйверами.

Другой вариант — HWiNFO. В этом мощном мониторе аппаратного обеспечения можно найти раздел, посвященный видеокарте, и посмотреть раздел Graphics Processor. Там будет указан Compute Capability (вычислительная способность), который напрямую связан с поддержкой версий CUDA.

Если карта физически поддерживает версию 8.6, но драйвер устарел, программа покажет аппаратные возможности, но при запуске софта возникнут ошибки.

import torch
print(torch.cuda.is_available())
print(torch.version.cuda)

from numba import cuda
print(cuda.is_available())