Svoj GPU NVIDIA môžete prechádzať v kontajneroch Docker a spúšťať programy CUDA na GPU NVIDIA z týchto kontajnerov Docker. Toto je veľmi užitočná funkcia na učenie sa AI (umelá inteligencia). Možnosť spúšťať kódy AI (t. j. Tensorflow) na kontajneroch Docker vám ušetrí veľa času. Môžete si tiež vyskúšať svoje kódy AI na rôznych verziách CUDA bez toho, aby ste museli meniť verziu CUDA vo svojom počítači.
Pohrávanie sa s ovládačmi NVIDIA a verziami CUDA vo vašom počítači môže spôsobiť problémy, ako napríklad nefunkčnosť ovládačov NVIDIA alebo zanechanie čiernej/modrej obrazovky smrti. Preto je vždy dobré nechať systém tak, ako je, a vykonať zmeny v izolovanom prostredí, ako sú kontajnery Docker. Váš hlavný počítač tak zostane čistý (od nepotrebných vývojových nástrojov).
V tomto článku vám ukážem, ako nastaviť Docker CE a NVIDIA docker na Ubuntu 22.04 LTS, aby ste mali prístup k NVIDIA GPU vášho počítača z kontajnerov Docker a spúšťali programy CUDA na vašom NVIDIA GPU.
Obsah:
- Kontrola, či sú na Ubuntu 22.04 nainštalované oficiálne ovládače GPU NVIDIA
- Inštalácia závislostí Docker CE na Ubuntu 22.04
- Inštalácia kľúčov Docker CE GPG na Ubuntu 22.04
- Inštalácia úložiska Docker CE na Ubuntu 22.04
- Inštalácia Docker CE na Ubuntu 22.04
- Pridanie prihlasovacieho používateľa Ubuntu 22.04 LTS do skupiny dockerov
- Kontrola, či bol Docker CE správne nainštalovaný na Ubuntu 22.04
- Inštalácia kľúčov GPG súpravy NVIDIA Container Toolkit na Ubuntu 22.04
- Inštalácia úložiska NVIDIA Container Toolkit na Ubuntu 22.04
- Inštalácia ovládačov nvidia-docker na Ubuntu 22.04
- Kontrola, či je GPU NVIDIA prístupné z kontajnerov Docker v Ubuntu 22.04
- Záver
- Referencie
Kontrola, či sú na Ubuntu 22.04 LTS nainštalované oficiálne ovládače GPU NVIDIA:
Skôr ako začnete, spustite nasledujúci príkaz, aby ste si overili, či máte v počítači nainštalovaný grafický procesor NVIDIA.
$ lspci | egrep -i 'vga|3d|display'
V tomto prípade mám v počítači nainštalovaný GPU NVIDIA GTX 1050 Ti. S najväčšou pravdepodobnosťou budete mať v počítači nainštalovanú inú GPU NVIDIA.
Tiež sa uistite, že máte v operačnom systéme Ubuntu 22.04 nainštalované oficiálne ovládače NVIDIA pomocou nasledujúceho príkazu:
$ lsmod | priľnavosť nvidia
Ak sú vo vašom operačnom systéme Ubuntu 22.04 nainštalované oficiálne ovládače NVIDIA, uvidíte nasledujúce výstupy.
Tiež skontrolujte, či oficiálne ovládače GPU NVIDIA fungujú s nasledujúcim príkazom:
Ak fungujú oficiálne ovládače GPU NVIDIA, uvidíte nasledujúce výstupy. Ako vidíte, na svojom počítači s Ubuntu 22.04 mám nainštalovaný oficiálny ovládač NVIDIA GPU Driver verzie 525.78.01.
Ak na svojom počítači Ubuntu 22.04 nemáte nainštalované oficiálne ovládače GPU NVIDIA a potrebujete s tým pomôcť, prečítajte si článok .
Inštalácia závislostí Docker CE na Ubuntu 22.04 LTS:
Pred inštaláciou Docker CE na Ubuntu 22.04 si budete musieť nainštalovať požadované balíčky závislosti Docker CE na Ubuntu 22.04.
Najprv aktualizujte vyrovnávaciu pamäť balíka APT pomocou nasledujúceho príkazu:
$ sudo apt update
Vyrovnávacia pamäť balíkov APT by sa mala aktualizovať.
Ak chcete nainštalovať požadované balíčky závislostí Docker CE, spustite nasledujúci príkaz:
Pre potvrdenie inštalácie stlačte A a potom stlačte
Mali by byť nainštalované požadované balíčky závislostí Docker CE.
Inštalácia kľúčov Docker CE GPG na Ubuntu 22.04 LTS:
V tejto časti vám ukážem, ako nainštalovať kľúč GPG úložiska balíkov Docker CE na Ubuntu 22.04.
Najprv vytvorte nový priečinok /etc/apt/kľúčenky s nasledujúcim príkazom:
$ sudo mkdir -p / atď / apt / kľúčenky
Ak chcete nainštalovať kľúč GPG úložiska balíkov Docker CE na Ubuntu 22.04, spustite nasledujúci príkaz:
$ zvlniť -fsSL https: // download.docker.com / linux / ubuntu / gpg | sudo gpg -- drahý -O / atď / apt / kľúčenky / docker.gpg
Inštalácia úložiska Docker CE na Ubuntu 22.04 LTS:
Ak chcete nainštalovať úložisko balíkov Docker CE na Ubuntu 22.04, spustite nasledujúci príkaz:
$ ozvena '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 / atď / apt / zdroje.zoznam.d / docker.list > / dev / nulový
Aktualizujte vyrovnávaciu pamäť balíkov APT, aby sa zmeny prejavili.
$ sudo apt update
Inštalácia Docker CE na Ubuntu 22.04 LTS:
Ak chcete nainštalovať najnovšiu verziu Docker CE na Ubuntu 22.04, spustite nasledujúci príkaz:
$ sudo apt-get nainštalovať docker-ce docker-ce-cli containerd.io docker-compose-plugin
Pre potvrdenie inštalácie stlačte A a potom stlačte
Docker CE a požadované balíčky závislostí sa sťahujú. Dokončenie bude chvíľu trvať.
Docker CE a požadované balíčky závislostí sa inštalujú. Dokončenie bude chvíľu trvať.
V tomto bode by ste mali nainštalovať Docker CE a požadované balíčky závislostí.
Pridanie prihlasovacieho používateľa Ubuntu 22.04 LTS do skupiny dockerov:
Ak chcete vytvoriť kontajnery Docker a spravovať ich bez použitia sudo alebo prihlásenia ako root, budete musieť pridať svojho prihlasovacieho používateľa do doker skupina.
Ak chcete pridať prihlasovacieho používateľa vášho Ubuntu 22.04 do doker skupina, spustite nasledujúci príkaz:
$ sudo usermod -aG docker $ ( kto som )
Aby sa zmeny prejavili, reštartujte počítač pomocou nasledujúceho príkazu:
$ sudo reštartovať
Kontrola, či bol Docker CE správne nainštalovaný na Ubuntu 22.04 LTS:
Po spustení počítača spustite nasledujúci príkaz a overte, či máte prístup k Dockeru bez oprávnení superužívateľa.
$ docker verzia
Ak všetko funguje, mali by ste vidieť nasledujúce výstupy.
Ako vidíte, používam Docker verziu 20.10.23 – najnovšiu verziu Docker CE v čase písania tohto článku.
Inštalácia kľúčov GPG súpravy NVIDIA Container Toolkit na Ubuntu 22.04:
V tejto časti vám ukážem, ako nainštalovať kľúč GPG úložiska balíkov NVIDIA Container Toolkit na Ubuntu 22.04.
Ak chcete nainštalovať kľúč GPG úložiska balíkov NVIDIA Container Toolkit na Ubuntu 22.04, spustite nasledujúci príkaz:
$ zvlniť -fsSL https: // nvidia.github.io / libnvidia-container / gpgkey | sudo gpg -- drahý -O / usr / zdieľam / kľúčenky / nvidia-container-toolkit-keyring.gpg
Inštalácia úložiska NVIDIA Container Toolkit na Ubuntu 22.04 LTS:
V tejto časti vám ukážem, ako nainštalovať úložisko balíkov NVIDIA Container Toolkit na Ubuntu 22.04.
Najprv vytvorte nový zdrojový súbor APT nvidia-container-toolkit.list v /etc/apt/sources.list.d/ adresár takto:
$ sudo nano / atď / apt / zdroje.zoznam.d / nvidia-container-toolkit.list
V nvidia-container-toolkit.list súbor, pridajte nasledujúci riadok a stlačte
Ak používate Ubuntu 20.04 LTS, potom všetko, čo musíte urobiť, je pridať nasledujúci riadok do nvidia-container-toolkit.list súbor a všetko ostatné by malo fungovať nezmenené.
Ak používate Ubuntu 18.04 LTS, potom všetko, čo musíte urobiť, je pridať nasledujúci riadok do nvidia-container-toolkit.list súbor a všetko ostatné by malo fungovať nezmenené.
Aby sa zmeny prejavili, aktualizujte vyrovnávaciu pamäť úložiska balíkov APT nasledujúcim príkazom:
Inštalácia ovládačov nvidia-docker na Ubuntu 22.04 LTS:
Ak chcete nainštalovať ovládače NVIDIA Docker na Ubuntu 22.04, spustite nasledujúci príkaz:
$ sudo apt Inštalácia nvidia-docker2
Pre potvrdenie inštalácie stlačte A a potom stlačte
Mali by byť nainštalované ovládače dokovacieho zariadenia NVIDIA.
Aby sa zmeny prejavili, reštartujte počítač pomocou nasledujúceho príkazu:
Kontrola, či je NVIDIA GPU prístupná z kontajnerov Docker v Ubuntu 22.04 LTS:
V tejto časti vám ukážem, ako vytvoriť Kontajner NVIDIA CUDA Docker a overte, či má kontajner prístup k GPU NVIDIA z vášho počítača.
Ak chcete vytvoriť kontajner NVIDIA CUDA 12 Docker založený na Ubuntu 20.04 LTS a spustiť nvidia-smi Po vytvorení príkazu v ňom skontrolujte, či má prístup k GPU NVIDIA z vášho počítača, spustite nasledujúci príkaz:
$ docker spustiť --rm --gpus všetky nvidia / cuda:12.0.0-base-ubuntu20.04 nvidia-smi
Docker ťahá nvidia/cuda:12.0.0-base-ubuntu20.04 obrázok z Docker Hub. Dokončenie bude chvíľu trvať.
Po stiahnutí obrazu dokovacieho zariadenia NVIDIA CUDA a vytvorení kontajnera sa nvidia-smi spustí sa na ňom príkaz a vytlačí výstup na konzole, ako môžete vidieť na obrázku nižšie.
Kontajner Docker používa Ovládač GPU NVIDIA 525.78.01 [1] a CUDA verzia 12.0 [2] . Ak vidíte podobné výstupy, kontajner Docker môže pristupovať k GPU NVIDIA vášho počítača.
Ak potrebujete použiť staršie verzie CUDA, skontrolujte .
Záver:
V tomto článku som vám ukázal, ako nastaviť úložisko balíkov Docker CE na Ubuntu 22.04. Tiež som vám ukázal, ako nainštalovať najnovšiu verziu Docker CE na Ubuntu 22.04. Ukázal som vám, ako nainštalovať úložisko balíkov NVIDIA Container Toolkit na Ubuntu 22.04, ako aj ako nainštalovať ovládač NVIDIA Docker na Ubuntu 22.04. Nakoniec som vám ukázal, ako získať prístup k GPU NVIDIA vášho počítača z kontajnera Docker.