Установка драйверов NVIDIA и CUDA на Ubuntu/Kubuntu

[nsfrolov]

Данная инструкция описывает процесс установки драйверов видеокарт NVIDIA и CUDA на следующие операционные системы: Ubuntu 22.04, Ubuntu 24.04.

Подготовка системы

1. Обновляем систему:
   sudo apt update
   sudo apt upgrade -y
2. На Ubuntu 22.04 для видеокарт RTX 4xxx, 5xxx серии, A100 и H100 необходимо обновить версию ядра (рекомендуем это сделать и для более старых видеокарт):
   sudo apt install linux-generic-hwe-22.04

Установка драйверов Nvidia и CUDA
1) Устанавливаем компилятор gcc, необходимый для сборки CUDA:
sudo apt install gcc-12 g++-12

2) Качаем и устанавливаем драйвера и CUDA. Для Ubuntu 24.04 поправьте в пути wget значение ubuntu2204 на ubuntu2404:
wget https://developer.download.nvidia.co..._1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt update
sudo apt install cuda -y



3) Прописываем переменные окружения для обнаружения CUDA вашими фреймворками и приложениями в .bashrc:
echo 'export PATH="/sbin:/bin:/usr/sbin:/usr/bin:${PATH}:/usr/local/cuda/bin"' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64\${LD_LIBRARY_PATH:+:\${LD_LIBRARY_PATH}}' >> ~/.bashrc
source ~/.bashrc


4) Проверяем установку драйверов на видеокарту:
sudo nvidia-smi


5) Проверяем установку CUDA:
nvcc -V