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:
- Minimálně 20 GB volného místa na disku
- Doporučeno 2 GB RAM
- 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.