Могу ли я установить CUDA 11.4 на Ubuntu 24.04? Если да, то как?

Вопрос или проблема

$ nvidia-smi   
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.256.02   Версия драйвера: 470.256.02   Версия CUDA: 11.4     |
|-------------------------------+----------------------+----------------------+
| Имя GPU         Режим постоянства| Bus-Id        Disp.A | Нестабильная некорр. ECC |
| Вентилятор  Темп  Производ.  Использование/Мощн. |         Использование памяти | Загруженность GPU  Вычисл. М. |
|                               |                      |               MIG М. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ...  Выкл | 00000000:65:00.0  Вкл |                  N/A |
| 31%   43C    P8    28W / 250W |    479MiB /  6075MiB |     27%      По умолчанию |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

Ответ или решение

Установка CUDA 11.4 на Ubuntu 24.04: детальный гайд

Введение

С ростом популярности графических процессоров и вычислений с их использованием, устанавливать и конфигурировать CUDA важнее, чем когда-либо. Если у вас есть система на базе Ubuntu 24.04 и вы хотите установить CUDA версии 11.4, вы столкнетесь с несколькими вызовами, поскольку дистрибутив по умолчанию предлагает более новые версии. Тем не менее, в этой статье мы рассмотрим, как установить CUDA 11.4 в Ubuntu 24.04, учитывая наличие устаревшего драйвера NVIDIA (nvidia-driver-470).

Требования

Перед тем как начать, убедитесь, что у вас установлены следующие компоненты:

1. Операционная система: Ubuntu 24.04.
2. Драйвер NVIDIA: nvidia-driver-470. Вы можете использовать команду nvidia-smi, чтобы проверить версию драйвера.
3. Основные зависимости: Убедитесь, что у вас есть обновления пакетов и необходимо программное обеспечение, установленное на сервере.

Шаг 1: Удаление существующего CUDA

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

sudo apt purge nvidia-cuda-toolkit

Шаг 2: Установка необходимых зависимостей

Обновите списки пакетов и установите базовые зависимости, необходимые для сборки:

sudo apt update
sudo apt install build-essential dkms

Шаг 3: Загрузка CUDA 11.4

Поскольку официальный веб-сайт NVIDIA не предоставляет прямой ссылки для установки CUDA 11.4 на Ubuntu 24.04, вы можете загрузить соответствующий пакет .run для установки. Для этого выполните следующие команды:

1. Перейдите на страницу загрузок CUDA.
2. Выберите пакет для загрузки: UNIX -> LINUX -> x86_64 -> Ubuntu -> 20.04.
3. Скопируйте ссылку на загрузку и используйте wget, чтобы скачать ее:

wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01-1_amd64.deb

Шаг 4: Установка CUDA 11.4

Теперь, когда файл загружен, установите его с помощью следующих команд:

sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-4-local/7fa2af80.pub
sudo apt update
sudo apt install cuda-11-4

Учтите, что вы устанавливаете CUDA для Ubuntu 20.04. Ubuntu 24.04 должна поддерживать эту версию CUDA, но вы можете столкнуться с проблемами совместимости из-за различных библиотек.

Шаг 5: Настройка окружения

После завершения установки вам нужно добавить CUDA в ваш PATH. Откройте файл ~/.bashrc и добавьте следующие строки:

export PATH=/usr/local/cuda-11.4/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

Сохраните изменения и примените их командой:

source ~/.bashrc

Шаг 6: Проверка установки

Чтобы убедиться, что CUDA установлена правильно, выполните следующую команду:

nvcc -V

Эта команда должна вернуть версию CUDA 11.4, если установка прошла успешно.

Заключение

Таким образом, вы успешно установили CUDA 11.4 на Ubuntu 24.04. Теперь вы готовы к разработке и запуску программ, использующих вычисления на GPU. Обязательно следите за совместимостью и обновлениями драйверов, чтобы избежать проблем в будущем. Если возникнут сложности, ознакомьтесь с официальной документацией NVIDIA или задавайте вопросы на форумах сообщества.