TensorFlow může využívat CPU a GPU k výpočtu složitých výpočtů umělé inteligence (AI) a strojového učení (ML). TensorFlow může použít jakýkoli GPU NVIDIA s podporou CUDA k urychlení programů AI/ML. Pokud nemáte GPU s podporou CUDA, TensorFlow použije CPU pro kódy AI/ML. Bez akcelerace GPU se výkon TensorFlow sníží ve složitých programech AI/ML.
V tomto článku vám ukážeme, jak nainstalovat TensorFlow s akcelerací NVIDIA CUDA/cuDNN na Debian 12 “Bookworm”.
Téma obsahu:
- Kontrola, zda máte v počítači nainstalovaný grafický procesor NVIDIA
- Instalace Python 3 PIP a Python Venv na Debian 12
- Vytvoření virtuálního prostředí Python 3 pro TensorFlow
- Upgrade Python 3 PIP ve virtuálním prostředí Python 3
- Instalace TensorFlow s podporou akcelerace NVIDIA CUDA
- Instalace TensorRT na Debian 12
- Aktivace virtuálního prostředí TensorFlow Python 3
- Přístup k TensorFlow a kontrola, zda je k dispozici akcelerace NVIDIA GPU/CUDA
- Závěr
Kontrola, zda máte v počítači nainstalovaný grafický procesor NVIDIA
Aby TensorFlow urychlil programy AI s NVIDIA GPU/CUDA, musíte mít Ovladače GPU NVIDIA a NVIDIA CUDA a cuDNN nainstalován na vašem operačním systému Debian 12.
Pokud potřebujete pomoc s instalací ovladačů GPU NVIDIA na váš operační systém Debian 12, přečtěte si tento článek .
Pokud potřebujete pomoc s instalací ovladačů NVIDIA CUDA a cuDNN na váš operační systém Debian 12, přečtěte si tento článek .
Jakmile budete mít na svém systému Debian 12 nainstalované ovladače GPU NVIDIA, měl by být k dispozici příkaz „nvidia-smi“.
Moduly jádra NVIDIA by také měly být načteny do vašeho systému Debian 12.
Jakmile budete mít nainstalované ovladače NVIDIA CUDA, měli byste mít v systému Debian 12 k dispozici příkaz „nvcc“.
Instalace Python 3 PIP a Python Venv na Debian 12
Chcete-li nainstalovat TensorFlow na Debian 12, musíte mít nainstalovaný modul PIP Python 3 a virtuálního prostředí Python (venv).
Nejprve aktualizujte mezipaměť úložiště balíčků APT pomocí následujícího příkazu:
$ sudo apt aktualizace
Chcete-li nainstalovat Python 3 PIP a virtuální prostředí Python 3 (venv), spusťte následující příkaz:
$ sudo apt Nainstalujte python3-pip python3-venv python3-dev Pro potvrzení instalace stiskněte „Y“ a poté stiskněte
Python 3 PIP a Python 3 venv se instalují. Dokončení chvíli trvá.
V tomto okamžiku by měly být nainstalovány Python 3 PIP a Python 3 venv.
Vytvoření virtuálního prostředí Python 3 pro TensorFlow
Standardní praxí pro instalaci knihoven Python na Debian 12 je jejich instalace ve virtuálním prostředí Pythonu, aby nezasahovaly do systémových balíčků/knihoven Pythonu.
Chcete-li vytvořit nové virtuální prostředí Python 3 pro TensorFlow v adresáři „/opt/tensorflow“, spusťte následující příkaz:
$ sudo python3 -m venv / opt / tensorflowUpgrade Python 3 PIP ve virtuálním prostředí Python 3
Chcete-li upgradovat Python 3 PIP na nejnovější verzi ve virtuálním prostředí Python 3 „/opt/tensorflow“, spusťte následující příkaz:
$ sudo / opt / tensorflow / zásobník / pip Nainstalujte --vylepšit pip
Instalace TensorFlow s podporou akcelerace NVIDIA CUDA
Chcete-li nainstalovat TensorFlow s podporou akcelerace NVIDIA CUDA do virtuálního prostředí Pythonu „/opt/tensorflow“, spusťte následující příkaz:
$ sudo / opt / tensorflow / zásobník / pip Nainstalujte tensorflow [ a-cuda ]Instaluje se TensorFlow s akcelerací NVIDIA CUDA. Dokončení chvíli trvá.
V tomto okamžiku by měl být nainstalován TensorFlow s podporou akcelerace NVIDIA CUDA.
Instalace TensorRT na Debian 12
NVIDIA TensorRT ještě více optimalizuje výkon hlubokého učení TensorFlow. TensorRT můžete nainstalovat do virtuálního prostředí TensorFlow Python „/opt/tensorflow“ pomocí následujícího příkazu:
$ sudo / opt / tensorflow / zásobník / pip Nainstalujte tensorrtNVIDIA TensorRT se instaluje do virtuálního prostředí Pythonu. Dokončení chvíli trvá.
V tomto okamžiku by měla být nainstalována NVIDIA TensorRT.
Aktivace virtuálního prostředí TensorFlow Python 3
Chcete-li aktivovat virtuální prostředí TensorFlow Python „/opt/tensorflow“, spusťte následující příkaz:
$ . / opt / tensorflow / zásobník / aktivovatMělo by být aktivováno virtuální prostředí TensorFlow Python 3.
Přístup k TensorFlow a kontrola, zda je k dispozici akcelerace NVIDIA GPU/CUDA
Chcete-li otevřít interaktivní shell Python 3, spusťte následující příkaz:
$ python3Interaktivní shell Pythonu 3 by měl být otevřen.
Nejprve importujte TensorFlow s následujícím řádkem kódu:
$ import tensorflow tak jako tfPo importu TensorFlow můžete zkontrolovat číslo verze TensorFlow, kterou jste nainstalovali, pomocí následujícího řádku kódu. Jak můžete vidět, na našem systému Debian 12 máme nainstalovaný TensorFlow 2.13.1.
$ tf.__version__Chcete-li ověřit, že TensorFlow může používat GPU NVIDIA, který jste nainstalovali do počítače pro akceleraci CUDA, spusťte následující řádek kódu. Jak můžete vidět, naše GPU NVIDIA je přístupné z TensorFlow.
$ tisk ( tf.config.list_physical_devices ( 'GPU' ) )
Chcete-li ukončit interaktivní shell Pythonu, spusťte následující řádek kódu:
$ přestat ( )Závěr
V tomto článku jsme vám ukázali, jak nainstalovat virtuální prostředí Python 3 PIP a Python 3 (venv) na Debian 12. Také jsme vám ukázali, jak vytvořit virtuální prostředí Python 3 pro TensorFlow na Debian 12 a jak nainstalovat TensorFlow pomocí NVIDIA Podpora akcelerace GPU/CUDA a NVIDIA TensorRT také na Debianu 12. Nakonec jsme vám ukázali, jak aktivovat virtuální prostředí TensorFlow Python a přistupovat k TensorFlow na Debianu 12.