TensorFlow pot utilitzar la CPU i la GPU per calcular els càlculs complexos d'intel·ligència artificial (IA) i d'aprenentatge automàtic (ML). TensorFlow pot utilitzar qualsevol GPU NVIDIA compatible amb CUDA per accelerar els programes AI/ML. Si no teniu una GPU compatible amb CUDA, TensorFlow utilitzarà la CPU per als codis AI/ML. Sense l'acceleració de la GPU, el rendiment de TensorFlow es degradarà en programes complexos d'IA/ML.
En aquest article, us mostrarem com instal·lar TensorFlow amb l'acceleració NVIDIA CUDA/cuDNN a Debian 12 'Bookworm'.
Tema de continguts:
- Comprovació de si teniu instal·lada la GPU NVIDIA al vostre ordinador
- Instal·lant Python 3 PIP i Python Venv a Debian 12
- Creació d'un entorn virtual Python 3 per a TensorFlow
- Actualització de Python 3 PIP a l'entorn virtual de Python 3
- Instal·lació de TensorFlow amb NVIDIA CUDA Acceleration Support
- Instal·lant TensorRT a Debian 12
- Activació de l'entorn virtual de TensorFlow Python 3
- Accedint a TensorFlow i comprovant si l'acceleració GPU/CUDA de NVIDIA està disponible
- Conclusió
Comprovació de si teniu instal·lada la GPU NVIDIA al vostre ordinador
Perquè TensorFlow acceleri els programes d'IA amb NVIDIA GPU/CUDA, heu de tenir el Controladors de GPU NVIDIA i NVIDIA CUDA i cuDNN instal·lat al vostre sistema operatiu Debian 12.
Si necessiteu ajuda per instal·lar els controladors de GPU de NVIDIA al vostre sistema operatiu Debian 12, llegiu aquest article .
Si necessiteu ajuda per instal·lar els controladors NVIDIA CUDA i cuDNN al vostre sistema operatiu Debian 12, llegiu aquest article .
Un cop tingueu els controladors de GPU NVIDIA instal·lats al vostre sistema Debian 12, l'ordre 'nvidia-smi' hauria d'estar disponible.
Els mòduls del nucli de NVIDIA també s'han de carregar al vostre sistema Debian 12.
Un cop tingueu instal·lats els controladors NVIDIA CUDA, hauríeu de tenir l'ordre 'nvcc' disponible al vostre sistema Debian 12.
Instal·lant Python 3 PIP i Python Venv a Debian 12
Per instal·lar TensorFlow a Debian 12, heu de tenir instal·lats el mòdul PIP de Python 3 i l'entorn virtual (venv) de Python.
Primer, actualitzeu la memòria cau del dipòsit de paquets APT amb l'ordre següent:
$ sudo actualització apta
Per instal·lar Python 3 PIP i l'entorn virtual Python 3 (venv), executeu l'ordre següent:
$ sudo apt instal·lar python3-pip python3-venv python3-dev Per confirmar la instal·lació, premeu 'Y' i després premeu
S'estan instal·lant Python 3 PIP i Python 3 venv. Es triga una estona a completar-se.
En aquest punt, s'han d'instal·lar Python 3 PIP i Python 3 venv.
Creació d'un entorn virtual Python 3 per a TensorFlow
La pràctica estàndard per instal·lar les biblioteques de Python a Debian 12 és instal·lar-les en un entorn virtual de Python perquè no interfereixin amb els paquets/biblioteques Python del sistema.
Per crear un nou entorn virtual Python 3 per a TensorFlow al directori '/opt/tensorflow', executeu l'ordre següent:
$ sudo python3 -m venv / optar / tensorflowActualització de Python 3 PIP a l'entorn virtual de Python 3
Per actualitzar Python 3 PIP a la darrera versió a l'entorn virtual de Python 3 '/opt/tensorflow', executeu l'ordre següent:
$ sudo / optar / tensorflow / paperera / pip instal·lar --actualització pip
Instal·lació de TensorFlow amb NVIDIA CUDA Acceleration Support
Per instal·lar TensorFlow amb suport d'acceleració NVIDIA CUDA a l'entorn virtual Python '/opt/tensorflow', executeu l'ordre següent:
$ sudo / optar / tensorflow / paperera / pip instal·lar tensorflow [ i-cuda ]S'està instal·lant TensorFlow amb acceleració NVIDIA CUDA. Es triga una estona a completar-se.
En aquest punt, s'hauria d'instal·lar TensorFlow amb suport d'acceleració NVIDIA CUDA.
Instal·lant TensorRT a Debian 12
NVIDIA TensorRT optimitza encara més el rendiment de l'aprenentatge profund de TensorFlow. Podeu instal·lar TensorRT a l'entorn virtual de TensorFlow Python '/opt/tensorflow' amb l'ordre següent:
$ sudo / optar / tensorflow / paperera / pip instal·lar tensorrtNVIDIA TensorRT s'està instal·lant a l'entorn virtual Python. Es triga una estona a completar-se.
En aquest punt, s'hauria d'instal·lar NVIDIA TensorRT.
Activació de l'entorn virtual de TensorFlow Python 3
Per activar l'entorn virtual de TensorFlow Python '/opt/tensorflow', executeu l'ordre següent:
$ . / optar / tensorflow / paperera / activarL'entorn virtual TensorFlow Python 3 hauria d'estar activat.
Accedint a TensorFlow i comprovant si l'acceleració GPU/CUDA de NVIDIA està disponible
Per obrir l'intèrpret d'ordres interactiu Python 3, executeu l'ordre següent:
$ python3S'hauria d'obrir l'intèrpret d'ordres interactiu Python 3.
Primer, importeu TensorFlow amb la següent línia de codi:
$ importar tensorflow com tfUn cop importat TensorFlow, podeu comprovar el número de versió de TensorFlow que heu instal·lat amb la següent línia de codi. Com podeu veure, tenim TensorFlow 2.13.1 instal·lat al nostre sistema Debian 12.
$ tf.__versió__Per verificar que TensorFlow pot utilitzar la GPU NVIDIA que heu instal·lat al vostre ordinador per a l'acceleració CUDA, executeu la línia de codi següent. Com podeu veure, la nostra GPU NVIDIA és accessible des de TensorFlow.
$ imprimir ( tf.config.list_physical_devices ( 'GPU' ) )
Per sortir de l'intèrpret d'ordres interactiu de Python, executeu la següent línia de codi:
$ sortir ( )Conclusió
En aquest article, us vam mostrar com instal·lar l'entorn virtual Python 3 PIP i Python 3 (venv) a Debian 12. També us vam mostrar com crear un entorn virtual Python 3 per a TensorFlow a Debian 12 i com instal·lar TensorFlow amb NVIDIA Suport d'acceleració GPU/CUDA i NVIDIA TensorRT també a Debian 12. Finalment, us vam mostrar com activar l'entorn virtual TensorFlow Python i accedir a TensorFlow a Debian 12.