Jak nainstalovat CUDA na Ubuntu 20.04 LTS

V tomto tutoriálu vám ukážeme, jak nainstalovat CUDA na Ubuntu 20.04 LTS. Pro ty z vás, kteří to nevěděli, CUDA znamená Compute Unified Device Architecture. CUDA obsahuje sadu nástrojů CUDA ( kompilátor, profil a ladicí program ), softwarový ovladač a CUDA SDK. Softwarová vrstva poskytuje přímý přístup k sadě virtuálních instrukcí GPU a paralelním výpočetním prvkům. Pro hluboké učení výzkumů a vývojářů frameworku použijte cuDNN pro vysoce výkonnou akceleraci GPU .

Tento článek předpokládá, že máte alespoň základní znalosti Linuxu, víte, jak používat shell, a co je nejdůležitější, hostujete svůj web na vlastním VPS. Instalace je poměrně jednoduchá a předpokládá, že běží v účtu root, pokud ne, možná budete muset přidat 'sudo ‘ k příkazům pro získání oprávnění root. Ukážu vám krok za krokem instalaci CUDA na Ubuntu 20.04 (Focal Fossa). Můžete postupovat podle stejných pokynů pro Ubuntu 18.04, 16.04 a jakoukoli jinou distribuci založenou na Debianu, jako je Linux Mint.

Předpoklady

• Server s jedním z následujících operačních systémů:Ubuntu 20.04, 18.04 a jakoukoli jinou distribucí založenou na Debianu, jako je Linux Mint.
• Abyste předešli případným problémům, doporučujeme použít novou instalaci operačního systému.
• Přístup SSH k serveru (nebo stačí otevřít Terminál, pokud jste na počítači).
• non-root sudo user nebo přístup k root user . Doporučujeme jednat jako non-root sudo user , protože však můžete poškodit svůj systém, pokud nebudete při jednání jako root opatrní.

Nainstalujte CUDA na Ubuntu 20.04 LTS Focal Fossa

Krok 1. Nejprve se ujistěte, že všechny vaše systémové balíčky jsou aktuální, spuštěním následujícího apt příkazy v terminálu.

sudo apt update
sudo apt upgrade

Krok 2. Instalace CUDA na Ubuntu 20.04.

Ve výchozím nastavení není balíček CUDA dostupný v základním úložišti Ubuntu 20.04. Nyní přidáme pin úložiště CUDA do vašeho systému pomocí následujícího příkazu:

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

Dále přesuneme špendlík do /etc/preferences.d adresář a přejmenujte jej na cuda-repository-pin-600 :

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

Poté stáhněte soubor CUDA Ubuntu pomocí následujícího příkazu:

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

Potom použijte dpkg příkaz ke spuštění souboru:

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

Nakonec balíček obsahuje a nainstaluje softwarový balíček CUDA:

sudo apt update
sudo apt install cuda

Krok 3. Instalace CuDNN.

Nyní stahujeme CuDNN nainstalovaný z oficiální stránky. Po registraci do NVIDIA Developer Program si můžete stáhnout nejnovější verzi cuDNN:

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

Dále zkopírujte následující soubory do adresáře 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*

Blahopřejeme! Úspěšně jste nainstalovali CUDA. Děkujeme, že jste použili tento návod k instalaci CUDA NVIDIA na systém Ubuntu 20.04 LTS Focal Fossa. Pro další nápovědu nebo užitečné informace vám doporučujeme zkontrolovat oficiální NVIDIA webové stránky.