Jak nainstalovat CUDA na Ubuntu 20.04

CUDA je paralelní výpočetní platforma a programovací model, který poskytuje pozoruhodný uživatelský zážitek při využití GPU pro každodenní univerzální výpočty. CUDA znamená C vypočítejte U nified D zařízení A architektura. Je vytvořen společností NVIDIA.

CUDA obsahuje sadu nástrojů CUDA (překladač, profil a ladicí program), softwarový ovladač a sadu CUDA SDK. Softwarová vrstva poskytuje přímý přístup k virtuální instrukční sadě GPU a paralelním výpočetním prvkům. Pro výzkumy hlubokého učení a vývojáři rámců použijte cuDNN pro vysoce výkonnou akceleraci GPU. Je to knihovna primitiv pro hluboké neuronové sítě.

V této příručce se naučíme, jak nainstalovat CUDA a CuDNN na Ubuntu 20.04 . Existují dva způsoby, jak toho dosáhnout. CUDA můžete nainstalovat z úložiště Ubuntu – což je nejjednodušší přístup – nebo nainstalovat z úložiště CUDA, které poskytuje nejnovější verzi CUDA.

Předpoklady:

1. Minimálně 20 GB volného místa na disku
2. Doporučeno 2 GB RAM
3. Měl by mít nainstalovaný grafický ovladač NVIDIA. Ovladač dodávaný s CUDA nemusí být nejnovější verze.

Metoda 1:Nainstalujte CUDA z úložiště Ubuntu

Začněte aktualizací seznamů balíčků ve vaší instanci Ubuntu 20.04.

$ sudo apt update

Dále nainstalujte sadu nástrojů CUDA pomocí správce balíčků APT následovně. Všimněte si, že instalace je náročná, a proto zajistěte rychlé a stabilní připojení k internetu a minimálně 10G místa na disku.

$ sudo apt install nvidia-cuda-toolkit

Jakmile je instalace dokončena, musíme přidat CUDA do PATH, abychom informovali shell o umístění CUDA. K tomu uvedeme PATH v souboru .bashrc. Otevřete tedy soubor pomocí textového editoru příkazového řádku.

$ sudo vim ~/.bashrc

Na úplný konec souboru zkopírujte a vložte následující řádky kódu.

# 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

Dále znovu načtěte změny v souboru .bashrc.

$ source ~/.bashrc

Nyní zkontrolujte a potvrďte nainstalovanou verzi CUDA pomocí následujícího příkazu.

$ nvcc --version

Metoda 2:Instalace CUDA z úložiště CUDA

Jak již bylo zmíněno dříve, úložiště CUDA poskytuje nejnovější verzi CUDA. Pokud se tedy chystáte nainstalovat nejnovější verzi, postupujte podle pokynů zde uvedených.

Nejprve si stáhněte pin úložiště CUDA:

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

Dále přesuňte š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é si stáhněte soubor CUDA Debian. Upozorňujeme, že soubor je poměrně velký – přesněji 2,3G v době psaní této příručky. Ujistěte se, že máte dobré připojení k internetu a dostatek místa na disku.

$ 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

Po stažení použijte dpkg nástroj ke spuštění souboru.

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

Dále nainstalujte klíč CUDA GPG následovně:

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

Nakonec aktualizujte seznamy balíčků a nainstalujte CUDA pomocí správce balíčků APT.

$ sudo apt update

$ sudo apt install cuda -y

Nainstalujte CuDNN

CuDNN nepřichází s Cudou. Chcete-li si stáhnout CuDNN, musíte se zaregistrovat a stát se členem NVIDIA Developer Program, který je zdarma.

Po registraci do NVIDIA Developer Program si můžete stáhnout nejnovější verzi cuDNN. Zde stahuji knihovnu cuDNN pro Linux.

Nejprve extrahujte balíček cuDNN:

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

Poté 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*

Závěr

V tomto tutoriálu jsme se naučili, jak nainstalovat CUDA a cuDNN na Ubuntu 20.04. Děkujeme za přečtení, zanechte prosím svůj názor a návrhy.