TensorFlow saab kasutada CPU-d ja GPU-d tehisintellekti (AI) ja masinõppe (ML) keerukate arvutuste arvutamiseks. TensorFlow saab AI/ML-programmide kiirendamiseks kasutada mis tahes CUDA-toega NVIDIA GPU-d. Kui teil pole CUDA-toega GPU-d, kasutab TensorFlow AI/ML-koodide jaoks CPU-d. Ilma GPU kiirenduseta halveneb TensorFlow jõudlus keerulistes AI/ML programmides.
Selles artiklis näitame teile, kuidas installida TensorFlow koos NVIDIA CUDA/cuDNN kiirendusega Debian 12 'Bookworm'.
Sisu teema:
- Kontrollige, kas teie arvutisse on installitud NVIDIA GPU
- Python 3 PIP ja Python Venv installimine Debian 12-sse
- Python 3 virtuaalse keskkonna loomine TensorFlow jaoks
- Python 3 PIP uuendamine Python 3 virtuaalkeskkonnas
- TensorFlow installimine NVIDIA CUDA kiirendustoega
- TensorRT installimine Debian 12-sse
- TensorFlow Python 3 virtuaalse keskkonna aktiveerimine
- Juurdepääs TensorFlow'le ja kontrollimine, kas NVIDIA GPU/CUDA kiirendus on saadaval
- Järeldus
Kontrollige, kas teie arvutisse on installitud NVIDIA GPU
Selleks, et TensorFlow saaks AI-programme NVIDIA GPU/CUDA abil kiirendada, peab teil olema NVIDIA GPU draiverid ja NVIDIA CUDA ja cuDNN installitud teie operatsioonisüsteemi Debian 12.
Kui vajate abi NVIDIA GPU draiverite installimisel operatsioonisüsteemi Debian 12, lugege seda artiklit .
Kui vajate abi NVIDIA CUDA ja cuDNN draiverite installimisel operatsioonisüsteemi Debian 12, lugege seda artiklit .
Kui olete oma Debian 12 süsteemi installinud NVIDIA GPU draiverid, peaks käsk 'nvidia-smi' olema saadaval.
NVIDIA kerneli moodulid tuleks laadida ka teie Debian 12 süsteemi.
Kui olete NVIDIA CUDA draiverid installinud, peaks teie Debian 12 süsteemis olema käsk 'nvcc'.
Python 3 PIP ja Python Venv installimine Debian 12-sse
TensorFlow installimiseks Debian 12-le peab teil olema Python 3 PIP ja Python virtuaalse keskkonna (venv) moodul.
Esmalt värskendage APT pakettide hoidla vahemälu järgmise käsuga:
$ sudo sobiv värskendus
Python 3 PIP ja Python 3 virtuaalse keskkonna (venv) installimiseks käivitage järgmine käsk:
$ sudo asjakohane installida python3-pip python3-venv python3-dev Installimise kinnitamiseks vajutage 'Y' ja seejärel vajutage
Python 3 PIP ja Python 3 venv installitakse. Selle valmimine võtab natuke aega.
Sel hetkel tuleks installida Python 3 PIP ja Python 3 venv.
Python 3 virtuaalse keskkonna loomine TensorFlow jaoks
Pythoni teekide installimise tavapärane praktika Debian 12-le on nende installimine Pythoni virtuaalsesse keskkonda, nii et need ei segaks süsteemi Pythoni pakette/teeke.
TensorFlow jaoks uue Python 3 virtuaalse keskkonna loomiseks kataloogis '/opt/tensorflow' käivitage järgmine käsk:
$ sudo python3 -m venv / opt / tensorflowPython 3 PIP uuendamine Python 3 virtuaalkeskkonnas
Python 3 PIP uuendamiseks Python 3 virtuaalkeskkonnas '/opt/tensorflow' uusimale versioonile käivitage järgmine käsk:
$ sudo / opt / tensorflow / prügikast / pip installida -- uuendada pip
TensorFlow installimine NVIDIA CUDA kiirendustoega
NVIDIA CUDA kiirendustoega TensorFlow installimiseks Pythoni virtuaalsesse keskkonda '/opt/tensorflow' käivitage järgmine käsk:
$ sudo / opt / tensorflow / prügikast / pip installida tensorflow [ ja-cuda ]Installimisel on NVIDIA CUDA kiirendusega TensorFlow. Selle valmimine võtab natuke aega.
Sel hetkel tuleks installida TensorFlow koos NVIDIA CUDA kiirenduse toega.
TensorRT installimine Debian 12-sse
NVIDIA TensorRT optimeerib TensorFlow süvaõppe jõudlust veelgi. TensorRT saate installida TensorFlow Pythoni virtuaalsesse keskkonda “/opt/tensorflow” järgmise käsuga:
$ sudo / opt / tensorflow / prügikast / pip installida tensorrtNVIDIA TensorRT installitakse Pythoni virtuaalsesse keskkonda. Selle valmimine võtab natuke aega.
Sel hetkel tuleks installida NVIDIA TensorRT.
TensorFlow Python 3 virtuaalse keskkonna aktiveerimine
TensorFlow Pythoni virtuaalse keskkonna '/opt/tensorflow' aktiveerimiseks käivitage järgmine käsk:
$ . / opt / tensorflow / prügikast / aktiveeridaTensorFlow Python 3 virtuaalne keskkond peaks olema aktiveeritud.
Juurdepääs TensorFlow'le ja kontrollimine, kas NVIDIA GPU/CUDA kiirendus on saadaval
Python 3 interaktiivse kesta avamiseks käivitage järgmine käsk:
$ python3Python 3 interaktiivne kest tuleks avada.
Esiteks importige TensorFlow järgmise koodireaga:
$ import tensorflow nagu tfKui TensorFlow on imporditud, saate kontrollida installitud TensorFlow versiooninumbrit järgmise koodireaga. Nagu näete, on meie Debian 12 süsteemi installitud TensorFlow 2.13.1.
$ tf.__version__Kontrollimaks, kas TensorFlow saab CUDA kiirendamiseks kasutada teie arvutisse installitud NVIDIA GPU-d, käivitage järgmine koodirida. Nagu näete, on meie NVIDIA GPU juurdepääsetav TensorFlow kaudu.
$ printida ( tf.config.list_physical_devices ( 'GPU' ) )
Pythoni interaktiivsest kestast väljumiseks käivitage järgmine koodirida:
$ lõpeta ( )Järeldus
Selles artiklis näitasime teile, kuidas installida Python 3 PIP ja Python 3 virtuaalne keskkond (venv) Debian 12-sse. Samuti näitasime, kuidas luua Python 3 virtuaalne keskkond TensorFlow jaoks Debian 12-s ja kuidas installida TensorFlow koos NVIDIAga. GPU/CUDA kiirenduse tugi ja NVIDIA TensorRT ka Debian 12-s. Lõpuks näitasime teile, kuidas aktiveerida TensorFlow Pythoni virtuaalne keskkond ja pääseda juurde TensorFlow'le Debian 12-s.