A TensorFlow CPU-t és GPU-t használhat a mesterséges intelligencia (AI) és a gépi tanulás (ML) komplex számításainak kiszámításához. A TensorFlow bármilyen CUDA által támogatott NVIDIA GPU-t használhat az AI/ML programok felgyorsításához. Ha nem rendelkezik CUDA-támogatott GPU-val, a TensorFlow a CPU-t fogja használni az AI/ML kódokhoz. GPU-gyorsítás nélkül a TensorFlow teljesítménye romlik az összetett AI/ML programokban.
Ebben a cikkben megmutatjuk, hogyan telepítheti a TensorFlow-t NVIDIA CUDA/cuDNN gyorsítással a Debian 12 „Bookworm” rendszeren.
Tartalom témája:
- Ellenőrizze, hogy van-e NVIDIA GPU telepítve a számítógépére
- Python 3 PIP és Python Venv telepítése Debian 12 rendszeren
- Python 3 virtuális környezet létrehozása a TensorFlow számára
- A Python 3 PIP frissítése a Python 3 virtuális környezetben
- A TensorFlow telepítése az NVIDIA CUDA gyorsítási támogatással
- A TensorRT telepítése Debian 12-re
- A TensorFlow Python 3 virtuális környezet aktiválása
- A TensorFlow elérése és annak ellenőrzése, hogy elérhető-e az NVIDIA GPU/CUDA-gyorsítás
- Következtetés
Ellenőrizze, hogy van-e NVIDIA GPU telepítve a számítógépére
Ahhoz, hogy a TensorFlow felgyorsítsa az AI-programokat az NVIDIA GPU/CUDA segítségével, rendelkeznie kell NVIDIA GPU illesztőprogramok és NVIDIA CUDA és cuDNN telepítve van a Debian 12 operációs rendszerre.
Ha segítségre van szüksége az NVIDIA GPU illesztőprogramok telepítéséhez a Debian 12 operációs rendszeren, olvassa el ezt a cikket .
Ha segítségre van szüksége az NVIDIA CUDA és cuDNN illesztőprogramok telepítéséhez Debian 12 operációs rendszerére, olvassa el ezt a cikket .
Miután telepítette az NVIDIA GPU illesztőprogramokat a Debian 12 rendszerére, az „nvidia-smi” parancsnak elérhetőnek kell lennie.
Az NVIDIA kernelmodulokat a Debian 12 rendszerére is be kell tölteni.
Miután telepítette az NVIDIA CUDA illesztőprogramokat, az „nvcc” parancsnak elérhetőnek kell lennie a Debian 12 rendszeren.
Python 3 PIP és Python Venv telepítése Debian 12 rendszeren
A TensorFlow Debian 12 rendszerre történő telepítéséhez telepítenie kell a Python 3 PIP és Python virtuális környezet (venv) modult.
Először frissítse az APT csomagtár gyorsítótárát a következő paranccsal:
$ sudo találó frissítés
A Python 3 PIP és a Python 3 virtuális környezet (venv) telepítéséhez futtassa a következő parancsot:
$ sudo alkalmas telepítés python3-pip python3-venv python3-dev A telepítés megerősítéséhez nyomja meg az „Y” gombot, majd nyomja meg a gombot
A Python 3 PIP és a Python 3 venv telepítése folyamatban van. Eltart egy ideig, amíg elkészül.
Ezen a ponton telepíteni kell a Python 3 PIP-t és a Python 3 venv-t.
Python 3 virtuális környezet létrehozása a TensorFlow számára
A Python-könyvtárak Debian 12-re történő telepítésének szokásos gyakorlata Python virtuális környezetbe történő telepítése, hogy ne zavarják a rendszer Python-csomagjait/könyvtárait.
Ha új Python 3 virtuális környezetet szeretne létrehozni a TensorFlow számára az „/opt/tensorflow” könyvtárban, futtassa a következő parancsot:
$ sudo python3 -m venv / dönt / tenzoráramlásA Python 3 PIP frissítése a Python 3 virtuális környezetben
A Python 3 PIP frissítéséhez a Python 3 „/opt/tensorflow” virtuális környezet legújabb verziójára futtassa a következő parancsot:
$ sudo / dönt / tenzoráramlás / kuka / csipog telepítés --frissítés csipog
A TensorFlow telepítése az NVIDIA CUDA gyorsítási támogatással
A TensorFlow NVIDIA CUDA gyorsítási támogatással történő telepítéséhez Python „/opt/tensorflow” virtuális környezetben futtassa a következő parancsot:
$ sudo / dönt / tenzoráramlás / kuka / csipog telepítés tenzoráramlás [ és-cuda ]A TensorFlow NVIDIA CUDA-gyorsítással telepítése folyamatban van. Eltart egy ideig, amíg elkészül.
Ezen a ponton telepíteni kell az NVIDIA CUDA gyorsítási támogatással rendelkező TensorFlow-t.
A TensorRT telepítése Debian 12-re
Az NVIDIA TensorRT még jobban optimalizálja a TensorFlow mélytanulás teljesítményét. A TensorRT telepíthető a TensorFlow Python „/opt/tensorflow” virtuális környezetre a következő paranccsal:
$ sudo / dönt / tenzoráramlás / kuka / csipog telepítés tensorrtAz NVIDIA TensorRT telepítése folyamatban van a Python virtuális környezetre. Eltart egy ideig, amíg elkészül.
Ezen a ponton telepíteni kell az NVIDIA TensorRT-t.
A TensorFlow Python 3 virtuális környezet aktiválása
A TensorFlow Python „/opt/tensorflow” virtuális környezet aktiválásához futtassa a következő parancsot:
$ . / dönt / tenzoráramlás / kuka / aktiváljaA TensorFlow Python 3 virtuális környezetet aktiválni kell.
A TensorFlow elérése és annak ellenőrzése, hogy elérhető-e az NVIDIA GPU/CUDA-gyorsítás
A Python 3 interaktív shell megnyitásához futtassa a következő parancsot:
$ python3Meg kell nyitni a Python 3 interaktív shelljét.
Először importálja a TensorFlow-t a következő kódsorral:
$ import tensorflow mint tfA TensorFlow importálása után a következő kódsorral ellenőrizheti a telepített TensorFlow verziószámát. Amint láthatja, a TensorFlow 2.13.1 telepítve van a Debian 12 rendszerünkön.
$ tf.__verzió__Annak ellenőrzéséhez, hogy a TensorFlow tudja használni a számítógépére telepített NVIDIA GPU-t a CUDA-gyorsításhoz, futtassa a következő kódsort. Amint látja, NVIDIA GPU-nk elérhető a TensorFlow-ból.
$ nyomtatás ( tf.config.list_physical_devices ( 'GPU' ) )
A Python interaktív shellből való kilépéshez futtassa a következő kódsort:
$ Kilépés ( )Következtetés
Ebben a cikkben bemutattuk, hogyan telepítheti a Python 3 PIP és Python 3 virtuális környezetet (venv) a Debian 12 rendszeren. Azt is megmutattuk, hogyan hozhat létre Python 3 virtuális környezetet a TensorFlow számára Debian 12 rendszeren, és hogyan telepítheti a TensorFlow-t NVIDIA-val GPU/CUDA-gyorsítási támogatás és NVIDIA TensorRT a Debian 12-n is. Végül megmutattuk, hogyan aktiválhatja a TensorFlow Python virtuális környezetet és hogyan érheti el a TensorFlow-t a Debian 12 rendszeren.