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

Linux, Ubuntu

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

 

CUDA — это платформа параллельных вычислений и модель программирования, которая обеспечивает замечательное взаимодействие с пользователем при использовании GPU для повседневных вычислений общего назначения. CUDA расшифровывается как C ompute U nified D Evice A rchitecture. Он создан NVIDIA.

CUDA включает набор инструментов CUDA (компилятор, профиль и отладчик), драйвер программного обеспечения и CUDA SDK. Программный уровень предоставляет прямой доступ к виртуальному набору команд графического процессора и параллельным вычислительным элементам. Для исследований в области глубокого обучения и разработчиков фреймворков используйте cuDNN для высокопроизводительного ускорения графического процессора. Это библиотека примитивов для глубоких нейронных сетей.

В этом руководстве мы узнаем, как установить CUDA и CuDNN в Ubuntu 20.04 . Есть два способа сделать это. Вы можете установить CUDA из репозитория Ubuntu — что является наиболее простым подходом — или установить из репозитория CUDA, который предоставляет последнюю версию CUDA.

Предпосылки:

  1. Минимум 20 ГБ свободного дискового пространства
  2. Рекомендуется 2 ГБ ОЗУ
  3. Должен быть установлен графический драйвер NVIDIA. Драйвер, поставляемый с CUDA, может быть не последней версией.

Метод 1: установите CUDA из репозитория Ubuntu

Начните с обновления списков пакетов в вашем экземпляре Ubuntu 20.04.

$ sudo apt update

Затем установите набор инструментов CUDA с помощью диспетчера пакетов APT, как показано ниже. Обратите внимание, что установка является интенсивной, и поэтому убедитесь, что у вас есть быстрое и стабильное подключение к Интернету и минимум 10 ГБ дискового пространства.

$ sudo apt install nvidia-cuda-toolkit

После завершения установки нам нужно добавить CUDA в PATH, чтобы уведомить оболочку о местонахождении CUDA. Для этого мы укажем ПУТЬ в файле .bashrc. Итак, откройте файл с помощью текстового редактора командной строки.

$ sudo vim ~/.bashrc

В самом конце файла скопируйте и вставьте следующие строки кода.

# set PATH for cuda 10.1 installation

if [ -d "/usr/local/cuda-10.1/bin/" ]; then

export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

fi

Затем перезагрузите изменения в файле .bashrc.

$ source ~/.bashrc

Теперь проверьте и подтвердите установленную версию CUDA, используя следующую команду.

$ nvcc --version
Как установить CUDA в Ubuntu 20.04

Метод 2: установить CUDA из репозитория CUDA

Как упоминалось ранее, репозиторий CUDA предоставляет последнюю версию CUDA. Итак, если вы собираетесь установить последнюю версию, следуйте приведенным здесь инструкциям.

Сначала загрузите пин-код репозитория CUDA:

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

Затем переместите булавку в /etc/preferences.dкаталог и переименуйте ее как cuda-repository-pin-600.

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

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

После этого загрузите файл CUDA Debian. Обратите внимание, что файл довольно большой — 2,3 ГБ, если быть точным на момент написания этого руководства. Убедитесь, что у вас хорошее подключение к Интернету и достаточно места на диске.

$ 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 GPG следующим образом:

$ sudo apt-key add /var/cuda-repo-ubuntu2004-11-4-local/7fa2af80.pub

Наконец, обновите списки пакетов и установите CUDA с помощью диспетчера пакетов APT.

$ sudo apt update
$ sudo apt install cuda -y

Установить CuDNN

CuDNN не поставляется с Cuda. Чтобы загрузить CuDNN, вам необходимо зарегистрироваться, чтобы стать участником программы для разработчиков NVIDIA, которая является бесплатной.
Как установить CUDA в Ubuntu 20.04

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

Сначала извлеките пакет 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 и cuDNN в Ubuntu 20.04. Спасибо за чтение, оставляйте свои отзывы и предложения.