Это руководство поможет вам установить и настроить следующие компоненты:
- CUDA Toolkit 12.2 — набор инструментов для работы с GPU от NVIDIA.
- cuDNN 8.9.5 — библиотека для ускорения вычислений на GPU с использованием CUDA.
- Nvidia Video Codec SDK 11.1 — набор библиотек для аппаратного декодирования и кодирования видео с использованием GPU NVIDIA.
Перед установкой CUDA и других компонентов обновите систему и установите необходимые зависимости:
sudo apt update
sudo apt install -y build-essential pkg-config cmake libtool libc6 libc6-dev unzip wgetДля контроля приоритетов репозиториев загрузите pin-файл:
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pinПереместите его в директорию предпочтений apt:
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600Загрузите установочный .deb файл:
wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda-repo-ubuntu2204-12-2-local_12.2.0-530.30.02-1_amd64.debУстановите репозиторий:
sudo dpkg -i cuda-repo-ubuntu2204-12-2-local_12.2.0-530.30.02-1_amd64.debСкопируйте ключи репозитория:
sudo cp /var/cuda-repo-ubuntu2204-12-2-local/cuda-*-keyring.gpg /usr/share/keyrings/Обновите индекс пакетов:
sudo apt updateУстановите CUDA Toolkit 12.2:
sudo apt install -y cuda-toolkit-12-2Добавьте в .bashrc пути к CUDA Toolkit:
echo 'export PATH=/usr/local/cuda-12.2/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrcЧтобы завершить установку, перезагрузите систему:
sudo rebootПосле перезагрузки проверьте версию CUDA:
nvcc --versionОжидаемый результат должен содержать информацию о версии release 12.2.
Перейдите на сайт NVIDIA cuDNN и скачайте соответствующий архив для версии CUDA 12.2 и cuDNN 8.9.5.
После скачивания архива распакуйте его и переместите в соответствующие каталоги:
tar -xzvf cudnn-12.2-linux-x64-v8.9.5.0.tgz
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*Для проверки правильности установки выполните команду:
dpkg -l | grep cudnnПерейдите на сайт NVIDIA Video Codec SDK и скачайте архив SDK 11.1.
Распакуйте архив и переместите его содержимое в нужную директорию:
tar -xvzf NVIDIA_Video_Codec_SDK_11.1.0_Linux.tar.gz
sudo mv NVIDIA_Video_Codec_SDK /usr/local/Для правильной работы с CUDA, переместите заголовочные файлы из SDK в папку CUDA. Это нужно для того, чтобы CUDA смогла найти нужные файлы при компиляции.
sudo cp /usr/local/NVIDIA_Video_Codec_SDK/include/* /usr/local/cuda/include/Добавьте путь к SDK в .bashrc:
echo 'export NV_CODEC_SDK=/usr/local/NVIDIA_Video_Codec_SDK' >> ~/.bashrc
source ~/.bashrcДля проверки, что SDK установлен корректно, выполните:
ls $NV_CODEC_SDKВы должны увидеть список директорий и файлов SDK, включая примеры и библиотеки.