Как установить CUDA в Ubuntu 20.04 LTS

Как установить CUDA в Ubuntu 20.04 LTS

 

В этом руководстве мы покажем вам, как установить CUDA в Ubuntu 20.04 LTS. Для тех из вас, кто не знал, CUDA означает Compute Unified Device Architecture. CUDA включает набор инструментов CUDA (компилятор, профиль и отладчик), драйвер программного обеспечения и CUDA SDK. Программный уровень предоставляет прямой доступ к виртуальному набору команд графического процессора и параллельным вычислительным элементам. Для исследований в области глубокого обучения и разработчиков фреймворков используйте cuDNN для высокопроизводительного ускорения графического процессора.

В этой статье предполагается, что у вас есть хотя бы базовые знания Linux, вы знаете, как использовать оболочку, и, что наиболее важно, вы размещаете свой сайт на собственном VPS. Установка довольно проста и предполагает, что вы работаете с учетной записью root, в противном случае вам может потребоваться добавить ‘ sudo‘ к командам для получения привилегий root. Я покажу вам пошаговую установку CUDA на Ubuntu 20.04 (Focal Fossa). Вы можете следовать тем же инструкциям для Ubuntu 18.04, 16.04 и любого другого дистрибутива на основе Debian, например Linux Mint.

Установите CUDA на Ubuntu 20.04 LTS Focal Fossa

Шаг 1. Во-первых, убедитесь, что все ваши системные пакеты обновлены, выполнив следующие aptкоманды в терминале.

sudo apt update
sudo apt upgrade

Шаг 2. Установка CUDA в Ubuntu 20.04.

По умолчанию пакет CUDA недоступен в базовом репозитории Ubuntu 20.04. Теперь мы добавляем контакт репозитория CUDA в вашу систему, используя следующую команду:

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin

Затем мы перемещаем булавку в каталог и переименовываем ее как :/etc/preferences.dcuda-repository-pin-600

sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600

После этого загрузите файл CUDA Ubuntu, используя следующую команду:

wget https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

Затем используйте dpkgкоманду для выполнения файла:

sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

Наконец, список пакетов и установка программного пакета CUDA:

sudo apt update
sudo apt install cuda

Шаг 3. Установка CuDNN.

Теперь скачиваем установленный с официальной страницы CuDNN . После регистрации в программе для разработчиков NVIDIA вы можете скачать последнюю версию cuDNN:

tar -xzvf cudnn-10.2-linux-x64-v8.2.4.15.tgz

Затем скопируйте следующие файлы в каталог CUDA:

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*

Поздравляю! Вы успешно установили CUDA . Благодарим за использование этого руководства по установке CUDA NVIDIA в системе Ubuntu 20.04 LTS Focal Fossa. Для получения дополнительной помощи или полезной информации мы рекомендуем вам посетить официальный сайт NVIDIA .