Nejnovější verze TensorFlow nepodporuje akceleraci NVIDIA CUDA/CuDNN nativně v operačních systémech Windows 10/11. Pokud tedy chcete nastavit nejnovější verzi vývojového prostředí TensorFlow s akcelerací NVIDIA CUDA/cuDNN na Windows 10/11, musíte tak učinit prostřednictvím Windows Subsystem for Linux (WSL).
V tomto článku vám ukážeme, jak nainstalovat WSL na Windows 10/11 a získat k němu přístup. Ukážeme vám také, jak nainstalovat nejnovější verzi TensorFlow s podporou akcelerace NVIDIA CUDA/cuDNN na systému Ubuntu WSL ve Windows 10/11.
Téma obsahu:
- Instalace ovladačů GPU NVIDIA na Windows 10/111
- Instalace NVIDIA CUDA a cuDNN na Windows 10/11
- Instalace WSL na Windows 10/11
- Přístup k terminálu WSL Ubuntu Linux ve Windows 10/11
- Kontrola, zda systém Ubuntu WSL může přistupovat k GPU NVIDIA Windows 10/11
- Instalace Pythonu 3 PIP na systém Ubuntu WSL
- Upgrade Pythonu 3 PIP na systému Ubuntu WSL
- Instalace TensorFlow s podporou akcelerace NVIDIA CUDA/cuDNN na systému Ubuntu WSL
- Kontrola, zda akcelerace TensorFlow CUDA funguje na systému Ubuntu WSL
- Přístup k systému Ubuntu WSL pomocí kódu Visual Studio pro vývoj TensorFlow
- Závěr
Instalace ovladačů GPU NVIDIA na Windows 10/11
Aby TensorFlow získal přístup k NVIDIA GPU pro akceleraci CUDA/cuDNN v systému Ubuntu WSL ve Windows 10/11, musíte mít na svém počítači nainstalovaný NVIDIA GPU a nainstalovat ovladač NVIDIA GPU na Windows 10/11. Pokud máte v počítači nainstalovaný grafický procesor NVIDIA a potřebujete pomoc s instalací ovladače GPU NVIDIA v systému Windows 10/11, přečtěte si tento článek .
Instalace NVIDIA CUDA a cuDNN na Windows 10/11
Jakmile nainstalujete ovladače GPU NVIDIA do systému Windows 10/11, musíte nainstalovat NVIDIA CUDA a NVIDIA cuDNN pro zrychlení TensorFlow CUDA/cuDNN, aby fungovaly v systému Ubuntu WSL.
Pokud potřebujete pomoc s instalací NVIDIA CUDA na váš operační systém Windows 10/11, přečtěte si tento článek .
Pokud potřebujete pomoc s instalací NVIDIA cuDNN do operačního systému Windows 10/11, přečtěte si tento článek.
Instalace WSL na Windows 10/11
Chcete-li nainstalovat WSL na Windows 10/11, otevřete aplikaci Terminál a spusťte následující příkaz:
$ wsl – nainstalovat
Klikněte na „Ano“.
WSL se instaluje. Dokončení chvíli trvá.
Jakmile se zobrazí následující výzva, klikněte na „Ano“.
Instalace by měla pokračovat.
Probíhá instalace operačního systému Ubuntu Linux. Dokončení chvíli trvá.
POZNÁMKA: Ubuntu je výchozí operační systém Windows WSL.
V tomto okamžiku by měl být na vašem počítači s Windows 10/11 nainstalován systém Ubuntu Linux WSL.
Aby se změny projevily, restartujte počítač.
Jakmile se váš počítač nabootuje, mělo by se zobrazit okno Terminál s výzvou k nastavení prvního uživatele Ubuntu.
Zadejte jméno uživatele systému Ubuntu WSL a stiskněte < Vstupte >.
Zadejte přihlašovací heslo pro nového uživatele a stiskněte < Vstupte >.
Zadejte znovu přihlašovací heslo a stiskněte < Vstupte >.
Pro systém Ubuntu WSL by měl být vytvořen nový uživatelský účet a Ubuntu by mělo být připraveno k použití.
Přístup k terminálu WSL Ubuntu Linux ve Windows 10/11
Chcete-li získat přístup k terminálu systému Ubuntu Linux WSL, otevřete aplikaci Terminal ve Windows 10/11 a klikněte na > Ubuntu .
Terminál systému Ubuntu Linux WSL by měl být otevřen.
Kontrola, zda systém Ubuntu WSL může přistupovat k GPU NVIDIA Windows 10/11
Chcete-li zkontrolovat, zda má systém Ubuntu WSL přístup k GPU NVIDIA vašeho počítače se systémem Windows 10/11, spusťte následující příkaz z terminálu systému Ubuntu WSL:
$nvidia-smi
Pokud má systém Ubuntu WSL přístup k GPU NVIDIA vašeho počítače se systémem Windows 10/11, zobrazí se informace o využití vašeho GPU NVIDIA, jak je znázorněno na následujícím snímku obrazovky:
Instalace Pythonu 3 PIP na systém Ubuntu WSL
Chcete-li nainstalovat TensorFlow na systém Ubuntu WSL, musíte mít na systému Ubuntu WSL nainstalován Python 3 PIP. Python 3 PIP můžete nainstalovat na systém Ubuntu WSL z oficiálního úložiště balíčků Ubuntu.
Nejprve aktualizujte mezipaměť databáze balíčků APT pomocí následujícího příkazu:
$ aktualizace sudo apt
Chcete-li nainstalovat Python 3 PIP na systém Ubuntu WSL, spusťte následující příkaz:
$ sudo apt install python3-pip
Pro potvrzení instalace stiskněte „Y“ a poté stiskněte < Vstupte >.
Python 3 PIP se instaluje na systém Ubuntu WSL. Dokončení chvíli trvá.
V tomto okamžiku by měl být Python 3 PIP nainstalován na systému Ubuntu WSL.
Chcete-li zkontrolovat, zda je Python 3 PIP dostupný v systému Ubuntu WSL, spusťte následující příkaz:
$ pip – verze
Jak můžete vidět, na našem systému Ubuntu WSL máme nainstalovaný Python 3 PIP 22.0.2.
Upgrade Pythonu 3 PIP na systému Ubuntu WSL
Chcete-li nainstalovat nejnovější verzi TensorFlow, musíte mít na svém systému Ubuntu WSL nainstalovanou nejnovější verzi Pythonu 3 PIP.
Chcete-li nainstalovat nejnovější verzi TensorFlow, musíte mít na svém systému Ubuntu WSL nainstalovanou nejnovější verzi Pythonu 3 PIP.
$ pip install –upgrade pip
Python PIP by měl být aktualizován na nejnovější verzi (verze 23.2.1 v době psaní tohoto článku).
$ pip – verze
Instalace TensorFlow s podporou akcelerace NVIDIA CUDA/cuDNN na systému Ubuntu WSL
Chcete-li nainstalovat TensorFlow s podporou akcelerace NVIDIA CUDA/cuDNN v systému Ubuntu WSL vašeho Windows 10/11, spusťte následující příkaz:
$ pip install tensorflow[and-cuda]
TensorFlow s podporou NVIDIA CUDA/cuDNN a požadovanými závislostmi se stahuje a instaluje. Dokončení chvíli trvá.
V tomto okamžiku by měl být na systému Ubuntu WSL nainstalován TensorFlow s podporou NVIDIA CUDA/cuDNN.
Kontrola, zda akcelerace TensorFlow CUDA funguje na systému Ubuntu WSL
Chcete-li zkontrolovat, zda akcelerace TensorFlow CUDA funguje na systému Ubuntu WSL, otevřete interaktivní shell/interpret Pythonu 3 pomocí následujícího příkazu:
$ python3
Chcete-li importovat TensorFlow, spusťte následující řádek kódu:
$ import tensorflow jako tf
Chcete-li zkontrolovat, zda byl TensorFlow importován správně, vytiskněte číslo verze TensorFlow pomocí následujícího řádku kódu:
$ tf.__version__
Jak můžete vidět, na našem systému Ubuntu WSL máme nainstalovaný TensorFlow 2.14.0.
Chcete-li zkontrolovat, zda je vaše GPU NVIDIA k dispozici pro akceleraci TensorFlow CUDA, spusťte následující řádek kódu:
$ tf.config.list_physical_devices(‘GPU’)
Jak vidíte, pro TensorFlow je k dispozici GPU zařízení. TensorFlow tedy může použít GPU NVIDIA vašeho počítače pro akceleraci CUDA.
Chcete-li ukončit interaktivní shell/interpret Pythonu 3, spusťte následující řádek kódu:
$ quit()
Přístup k systému Ubuntu WSL pomocí kódu Visual Studio pro vývoj TensorFlow
Visual Studio Code je skvělý editor kódu pro vývoj TensorFlow. Pokud chcete získat přístup k systému Ubuntu WSL pomocí Visual Studio Code pro vývoj TensorFlow a potřebujete s tím pomoc, přečtěte si tento článek.
Závěr
V tomto článku jsme vám ukázali, jak nainstalovat Ubuntu Linux přes WSL na Windows 10/11. Také jsme vám ukázali, jak přistupovat k terminálu systému Ubuntu WSL na Windows 10/11 a jak nainstalovat nejnovější verzi TensorFlow s podporou akcelerace NVIDIA CUDA/cuDNN i na systém Ubuntu WSL.