GPU NVIDIA můžete procházet v kontejnerech Docker a spouštět programy CUDA na GPU NVIDIA z těchto kontejnerů Docker. Toto je velmi užitečná funkce pro výuku AI (Artificial Intelligence). Možnost spouštět kódy AI (tj. Tensorflow) na kontejnerech Docker vám ušetří spoustu času. Můžete si také vyzkoušet své kódy AI na různých verzích CUDA, aniž byste museli měnit verzi CUDA na vašem počítači.
Pohrávání si s ovladači NVIDIA a verzemi CUDA na vašem počítači může způsobit problémy, jako je nefunkčnost ovladačů NVIDIA nebo černá/modrá obrazovka smrti. Vždy je tedy dobré ponechat systém tak, jak je, a provádět změny v izolovaném prostředí, jako jsou kontejnery Docker. Váš hlavní počítač tak zůstane čistý (od zbytečných vývojářských nástrojů).
V tomto článku vám ukážu, jak nastavit Docker CE a NVIDIA docker na Ubuntu 22.04 LTS, abyste měli přístup k GPU NVIDIA vašeho počítače z kontejnerů Docker a spouštěli programy CUDA na GPU NVIDIA.
Obsah:
- Kontrola, zda jsou na Ubuntu 22.04 nainstalovány oficiální ovladače GPU NVIDIA
- Instalace závislostí Docker CE na Ubuntu 22.04
- Instalace klíčů Docker CE GPG na Ubuntu 22.04
- Instalace úložiště Docker CE na Ubuntu 22.04
- Instalace Docker CE na Ubuntu 22.04
- Přidání přihlašovacího uživatele Ubuntu 22.04 LTS do skupiny dockerů
- Kontrola, zda byl Docker CE správně nainstalován na Ubuntu 22.04
- Instalace klíčů GPG sady NVIDIA Container Toolkit na Ubuntu 22.04
- Instalace úložiště NVIDIA Container Toolkit na Ubuntu 22.04
- Instalace ovladačů nvidia-docker na Ubuntu 22.04
- Kontrola, zda je GPU NVIDIA přístupný z kontejnerů Docker v Ubuntu 22.04
- Závěr
- Reference
Kontrola, zda jsou na Ubuntu 22.04 LTS nainstalovány oficiální ovladače GPU NVIDIA:
Než začnete, spusťte následující příkaz, abyste ověřili, že máte v počítači nainstalovanou GPU NVIDIA.
$ lspci | egrep -i 'vga|3d|display'
V tomto případě mám v počítači nainstalovanou GPU NVIDIA GTX 1050 Ti. S největší pravděpodobností budete mít v počítači nainstalovanou jinou GPU NVIDIA.
Také se ujistěte, že máte v operačním systému Ubuntu 22.04 nainstalované oficiální ovladače NVIDIA pomocí následujícího příkazu:
$ lsmod | rukojeť nvidia
Pokud jsou na vašem operačním systému Ubuntu 22.04 nainstalovány oficiální ovladače NVIDIA, uvidíte následující výstupy.
Také zkontrolujte, zda oficiální ovladače GPU NVIDIA fungují s následujícím příkazem:
Pokud oficiální ovladače GPU NVIDIA fungují, uvidíte následující výstupy. Jak můžete vidět, na svém počítači Ubuntu 22.04 mám nainstalovaný oficiální ovladač NVIDIA GPU Driver verze 525.78.01.
Pokud nemáte na svém počítači Ubuntu 22.04 nainstalované oficiální ovladače GPU NVIDIA a potřebujete s tím pomoc, přečtěte si článek .
Instalace závislostí Docker CE na Ubuntu 22.04 LTS:
Než budete moci nainstalovat Docker CE na Ubuntu 22.04, budete muset nainstalovat požadované balíčky závislostí Docker CE na Ubuntu 22.04.
Nejprve aktualizujte mezipaměť úložiště balíčků APT pomocí následujícího příkazu:
$ sudo apt aktualizace
Mezipaměť úložiště balíčků APT by měla být aktualizována.
Chcete-li nainstalovat požadované balíčky závislostí Docker CE, spusťte následující příkaz:
Pro potvrzení instalace stiskněte A a poté stiskněte
Měly by být nainstalovány požadované balíčky závislostí Docker CE.
Instalace klíčů Docker CE GPG na Ubuntu 22.04 LTS:
V této části vám ukážu, jak nainstalovat klíč GPG úložiště balíčků Docker CE na Ubuntu 22.04.
Nejprve vytvořte novou složku /etc/apt/klíčenky s následujícím příkazem:
$ sudo mkdir -p / atd / apt / klíčenky
Chcete-li nainstalovat klíč GPG úložiště balíčků Docker CE na Ubuntu 22.04, spusťte následující příkaz:
$ kučera -fsSL https: // download.docker.com / linux / ubuntu / gpg | sudo gpg --miláčku -Ó / atd / apt / klíčenky / docker.gpg
Instalace úložiště Docker CE na Ubuntu 22.04 LTS:
Chcete-li nainstalovat úložiště balíčků Docker CE na Ubuntu 22.04, spusťte následující příkaz:
$ echo 'deb [arch= $(dpkg --print-architecture) sign-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stabilní' | sudo tričko / atd / apt / zdroje.seznam.d / docker.list > / dev / nula
Aktualizujte mezipaměť úložiště balíčků APT, aby se změny projevily.
$ sudo apt aktualizace
Instalace Docker CE na Ubuntu 22.04 LTS:
Chcete-li nainstalovat nejnovější verzi Docker CE na Ubuntu 22.04, spusťte následující příkaz:
$ sudo instalace apt-get docker-ce docker-ce-cli containerd.io docker-compose-plugin
Pro potvrzení instalace stiskněte A a poté stiskněte
Docker CE a požadované balíčky závislostí se stahují. Dokončení bude chvíli trvat.
Docker CE a požadované balíčky závislostí se instalují. Dokončení bude chvíli trvat.
V tomto okamžiku by měl být nainstalován Docker CE a požadované balíčky závislostí.
Přidání přihlašovacího uživatele Ubuntu 22.04 LTS do skupiny dockerů:
Chcete-li vytvořit kontejnery Docker a spravovat je bez použití sudo nebo přihlášení jako uživatel root, budete muset přidat svého přihlašovacího uživatele do přístavní dělník skupina.
Chcete-li přidat přihlašovacího uživatele vašeho Ubuntu 22.04 do přístavní dělník group, spusťte následující příkaz:
$ sudo usermod -aG docker $ ( kdo jsem )
Aby se změny projevily, restartujte počítač pomocí následujícího příkazu:
$ sudo restartovat
Kontrola, zda byl Docker CE správně nainstalován na Ubuntu 22.04 LTS:
Po spuštění počítače spusťte následující příkaz, abyste ověřili, že máte přístup k Dockeru bez oprávnění superuživatele.
$ docker verze
Pokud vše funguje, měli byste vidět následující výstupy.
Jak můžete vidět, používám Docker verzi 20.10.23 – nejnovější verzi Docker CE v době psaní tohoto článku.
Instalace klíčů GPG sady NVIDIA Container Toolkit na Ubuntu 22.04:
V této části vám ukážu, jak nainstalovat klíč GPG z úložiště balíčků NVIDIA Container Toolkit na Ubuntu 22.04.
Chcete-li nainstalovat klíč GPG z úložiště balíčků NVIDIA Container Toolkit na Ubuntu 22.04, spusťte následující příkaz:
$ kučera -fsSL https: // nvidia.github.io / libnvidia-container / gpgkey | sudo gpg --miláčku -Ó / usr / podíl / klíčenky / nvidia-container-toolkit-keyring.gpg
Instalace úložiště NVIDIA Container Toolkit na Ubuntu 22.04 LTS:
V této části vám ukážu, jak nainstalovat úložiště balíčků NVIDIA Container Toolkit na Ubuntu 22.04.
Nejprve vytvořte nový zdrojový soubor APT nvidia-container-toolkit.list v /etc/apt/sources.list.d/ adresář takto:
$ sudo nano / atd / apt / zdroje.seznam.d / nvidia-container-toolkit.list
V nvidia-container-toolkit.list soubor, přidejte následující řádek a stiskněte
Pokud používáte Ubuntu 20.04 LTS, pak vše, co musíte udělat, je přidat následující řádek do nvidia-container-toolkit.list soubor a vše ostatní by mělo fungovat beze změny.
Pokud používáte Ubuntu 18.04 LTS, pak vše, co musíte udělat, je přidat následující řádek do nvidia-container-toolkit.list soubor a vše ostatní by mělo fungovat beze změny.
Aby se změny projevily, aktualizujte mezipaměť úložiště balíčků APT pomocí následujícího příkazu:
Instalace ovladačů nvidia-docker na Ubuntu 22.04 LTS:
Chcete-li nainstalovat ovladače NVIDIA Docker na Ubuntu 22.04, spusťte následující příkaz:
$ sudo apt Nainstalujte nvidia-docker2
Pro potvrzení instalace stiskněte A a poté stiskněte
Měly by být nainstalovány ovladače dokovací stanice NVIDIA.
Aby se změny projevily, restartujte počítač pomocí následujícího příkazu:
Kontrola, zda je GPU NVIDIA přístupný z kontejnerů Docker v Ubuntu 22.04 LTS:
V této sekci vám ukážu, jak vytvořit Kontejner NVIDIA CUDA Docker a ověřte, že kontejner má přístup k GPU NVIDIA z vašeho počítače.
Chcete-li vytvořit kontejner NVIDIA CUDA 12 Docker založený na Ubuntu 20.04 LTS a spustit nvidia-smi Jakmile bude vytvořen, spusťte následující příkaz, abyste ověřili, zda má přístup k GPU NVIDIA z vašeho počítače:
$ docker spustit --rm --gpus všechny nvidie / cuda:12.0.0-base-ubuntu20.04 nvidia-smi
Docker táhne nvidia/cuda:12.0.0-base-ubuntu20.04 obrázek z Docker Hub. Dokončení bude chvíli trvat.
Jakmile je natažen obraz dokovacího zařízení NVIDIA CUDA a vytvořen kontejner, nvidia-smi spustí se na něm příkaz a vytiskne výstup na konzoli, jak můžete vidět na obrázku níže.
Kontejner Docker používá Ovladač GPU NVIDIA 525.78.01 [1] a CUDA verze 12.0 [2] . Pokud vidíte podobné výstupy, může kontejner Docker přistupovat k GPU NVIDIA vašeho počítače.
Pokud potřebujete používat starší verze CUDA, zkontrolujte .
Závěr:
V tomto článku jsem vám ukázal, jak nastavit úložiště balíčků Docker CE na Ubuntu 22.04. Také jsem vám ukázal, jak nainstalovat nejnovější verzi Docker CE na Ubuntu 22.04. Ukázal jsem vám, jak nainstalovat úložiště balíčků NVIDIA Container Toolkit na Ubuntu 22.04 a také jak nainstalovat ovladač NVIDIA Docker na Ubuntu 22.04. Nakonec jsem vám ukázal, jak získat přístup k GPU NVIDIA vašeho počítače z kontejneru Docker.