Najnovšia verzia TensorFlow nepodporuje zrýchlenie NVIDIA CUDA/CuDNN natívne v operačných systémoch Windows 10/11. Ak teda chcete nastaviť najnovšiu verziu vývojového prostredia TensorFlow s akceleráciou NVIDIA CUDA/cuDNN v systéme Windows 10/11, musíte tak urobiť prostredníctvom podsystému Windows pre Linux (WSL).
V tomto článku vám ukážeme, ako nainštalovať WSL na Windows 10/11 a získať k nemu prístup. Ukážeme vám tiež, ako nainštalovať najnovšiu verziu TensorFlow s podporou akcelerácie NVIDIA CUDA/cuDNN v systéme Ubuntu WSL v systéme Windows 10/11.
Téma obsahu:
- Inštalácia ovládačov GPU NVIDIA v systéme Windows 10/111
- Inštalácia NVIDIA CUDA a cuDNN na Windows 10/11
- Inštalácia WSL na Windows 10/11
- Prístup k terminálu WSL Ubuntu Linux v systéme Windows 10/11
- Kontrola, či systém Ubuntu WSL môže pristupovať k GPU NVIDIA systému Windows 10/11
- Inštalácia Pythonu 3 PIP na systéme Ubuntu WSL
- Inovácia Python 3 PIP na systéme Ubuntu WSL
- Inštalácia TensorFlow s podporou akcelerácie NVIDIA CUDA/cuDNN na systéme Ubuntu WSL
- Kontrola, či akcelerácia TensorFlow CUDA funguje na systéme Ubuntu WSL
- Prístup k systému Ubuntu WSL s kódom Visual Studio pre vývoj TensorFlow
- Záver
Inštalácia ovládačov GPU NVIDIA v systéme Windows 10/11
Aby TensorFlow získal prístup k NVIDIA GPU pre akceleráciu CUDA/cuDNN v systéme Ubuntu WSL v systéme Windows 10/11, musíte mať vo svojom počítači nainštalovaný GPU NVIDIA a nainštalovať ovládač GPU NVIDIA v systéme Windows 10/11. Ak máte v počítači nainštalovaný grafický procesor NVIDIA a potrebujete pomoc pri inštalácii ovládača grafického procesora NVIDIA v systéme Windows 10/11, prečítajte si tento článok .
Inštalácia NVIDIA CUDA a cuDNN na Windows 10/11
Po nainštalovaní ovládačov GPU NVIDIA do systému Windows 10/11 musíte nainštalovať NVIDIA CUDA a NVIDIA cuDNN pre zrýchlenie TensorFlow CUDA/cuDNN, aby fungovali v systéme Ubuntu WSL.
Ak potrebujete pomoc pri inštalácii NVIDIA CUDA do operačného systému Windows 10/11, prečítajte si tento článok .
Ak potrebujete pomoc pri inštalácii NVIDIA cuDNN do operačného systému Windows 10/11, prečítajte si tento článok.
Inštalácia WSL na Windows 10/11
Ak chcete nainštalovať WSL v systéme Windows 10/11, otvorte aplikáciu Terminál a spustite nasledujúci príkaz:
$ wsl – inštalácia
Kliknite na „Áno“.
WSL sa inštaluje. Dokončenie chvíľu trvá.
Keď uvidíte nasledujúcu výzvu, kliknite na „Áno“.
Inštalácia by mala pokračovať.
Prebieha inštalácia operačného systému Ubuntu Linux. Dokončenie chvíľu trvá.
POZNÁMKA: Ubuntu je predvolený operačný systém Windows WSL.
V tomto bode by mal byť systém Ubuntu Linux WSL nainštalovaný na vašom počítači so systémom Windows 10/11.
Aby sa zmeny prejavili, reštartujte počítač.
Po spustení počítača by sa malo zobraziť okno terminálu s výzvou na nastavenie prvého používateľa Ubuntu.
Zadajte meno používateľa systému Ubuntu WSL a stlačte < Zadajte >.
Zadajte prihlasovacie heslo pre nového používateľa a stlačte < Zadajte >.
Znova zadajte prihlasovacie heslo a stlačte < Zadajte >.
Pre systém Ubuntu WSL by mal byť vytvorený nový používateľský účet a Ubuntu by malo byť pripravené na použitie.
Prístup k terminálu WSL Ubuntu Linux v systéme Windows 10/11
Ak chcete získať prístup k terminálu systému Ubuntu Linux WSL, otvorte aplikáciu Terminal v systéme Windows 10/11 a kliknite na > Ubuntu .
Mal by sa otvoriť terminál systému Ubuntu Linux WSL.
Kontrola, či systém Ubuntu WSL môže pristupovať k GPU NVIDIA systému Windows 10/11
Ak chcete skontrolovať, či má systém Ubuntu WSL prístup k GPU NVIDIA vášho počítača so systémom Windows 10/11, spustite nasledujúci príkaz z terminálu systému Ubuntu WSL:
$nvidia-smi
Ak má systém Ubuntu WSL prístup k GPU NVIDIA vášho počítača so systémom Windows 10/11, zobrazia sa informácie o používaní vášho GPU NVIDIA, ako je znázornené na nasledujúcej snímke obrazovky:
Inštalácia Pythonu 3 PIP na systéme Ubuntu WSL
Ak chcete nainštalovať TensorFlow na systém Ubuntu WSL, musíte mať nainštalovaný Python 3 PIP v systéme Ubuntu WSL. Python 3 PIP môžete nainštalovať na systém Ubuntu WSL z oficiálneho úložiska balíkov Ubuntu.
Najprv aktualizujte vyrovnávaciu pamäť databázy balíkov APT pomocou nasledujúceho príkazu:
$ sudo apt aktualizácia
Ak chcete nainštalovať Python 3 PIP na systém Ubuntu WSL, spustite nasledujúci príkaz:
$ sudo apt install python3-pip
Na potvrdenie inštalácie stlačte „Y“ a potom stlačte < Zadajte >.
Python 3 PIP sa inštaluje na systém Ubuntu WSL. Dokončenie chvíľu trvá.
V tomto bode by mal byť Python 3 PIP nainštalovaný na systéme Ubuntu WSL.
Ak chcete skontrolovať, či je Python 3 PIP dostupný v systéme Ubuntu WSL, spustite nasledujúci príkaz:
$ pip – verzia
Ako môžete vidieť, na našom systéme Ubuntu WSL máme nainštalovaný Python 3 PIP 22.0.2.
Inovácia Python 3 PIP na systéme Ubuntu WSL
Ak chcete nainštalovať najnovšiu verziu TensorFlow, musíte mať na svojom systéme Ubuntu WSL nainštalovanú najnovšiu verziu Python 3 PIP.
Ak chcete nainštalovať najnovšiu verziu TensorFlow, musíte mať na svojom systéme Ubuntu WSL nainštalovanú najnovšiu verziu Python 3 PIP.
$ pip install – upgrade pip
Python PIP by mal byť aktualizovaný na najnovšiu verziu (verzia 23.2.1 v čase písania tohto článku).
$ pip – verzia
Inštalácia TensorFlow s podporou akcelerácie NVIDIA CUDA/cuDNN na systéme Ubuntu WSL
Ak chcete nainštalovať TensorFlow s podporou akcelerácie NVIDIA CUDA/cuDNN v systéme Ubuntu WSL vášho Windows 10/11, spustite nasledujúci príkaz:
$ pip nainštalovať tensorflow[and-cuda]
Prebieha sťahovanie a inštalácia TensorFlow s podporou NVIDIA CUDA/cuDNN a požadovanými závislosťami. Dokončenie chvíľu trvá.
V tomto bode by mal byť na systéme Ubuntu WSL nainštalovaný TensorFlow s podporou NVIDIA CUDA/cuDNN.
Kontrola, či akcelerácia TensorFlow CUDA funguje na systéme Ubuntu WSL
Ak chcete skontrolovať, či akcelerácia TensorFlow CUDA funguje na systéme Ubuntu WSL, otvorte interaktívny shell/interpret Python 3 pomocou nasledujúceho príkazu:
$ python3
Ak chcete importovať TensorFlow, spustite nasledujúci riadok kódu:
$ importovať tensorflow ako tf
Ak chcete skontrolovať, či bol TensorFlow importovaný správne, vytlačte číslo verzie TensorFlow s nasledujúcim riadkom kódu:
$ tf.__version__
Ako vidíte, na našom systéme Ubuntu WSL máme nainštalovaný TensorFlow 2.14.0.
Ak chcete skontrolovať, či je vaša GPU NVIDIA k dispozícii pre akceleráciu TensorFlow CUDA, spustite nasledujúci riadok kódu:
$ tf.config.list_physical_devices(‘GPU’)
Ako vidíte, pre TensorFlow je k dispozícii GPU zariadenie. TensorFlow teda môže použiť GPU NVIDIA vášho počítača na zrýchlenie CUDA.
Ak chcete ukončiť interaktívny shell/interpret Python 3, spustite nasledujúci riadok kódu:
$ quit()
Prístup k systému Ubuntu WSL s kódom Visual Studio pre vývoj TensorFlow
Visual Studio Code je skvelý editor kódu pre vývoj TensorFlow. Ak chcete získať prístup k systému Ubuntu WSL s kódom Visual Studio pre vývoj TensorFlow a potrebujete s tým pomoc, prečítajte si tento článok.
Záver
V tomto článku sme vám ukázali, ako nainštalovať Ubuntu Linux cez WSL na Windows 10/11. Tiež sme vám ukázali, ako pristupovať k terminálu systému Ubuntu WSL na Windows 10/11 a ako nainštalovať najnovšiu verziu TensorFlow s podporou akcelerácie NVIDIA CUDA/cuDNN aj na systém Ubuntu WSL.