PyTorch pot accelerar les aplicacions AI/ML mitjançant una GPU NVIDIA mitjançant la biblioteca NVIDIA CUDA de manera nativa igual que TensorFlow.
En aquest article, us mostrarem com instal·lar PyTorch amb suport d'acceleració GPU/CUDA de NVIDIA a Debian 12 'Bookworm'.
Tema de continguts:
- Instal·lació dels controladors de la GPU de NVIDIA a Debian 12
- Instal·lació de NVIDIA CUDA a Debian 12
- Instal·lació de Python 3 PIP i Python 3 Virtual Environment (venv) a Debian 12
- Creació d'un entorn virtual Python 3 per a PyTorch
- Actualització de Python 3 PIP a la darrera versió a l'entorn virtual de Python 3 PyTorch
- Instal·lació de PyTorch amb NVIDIA GPU/CUDA Acceleration Support a Debian 12
- Activació de l'entorn virtual PyTorch Python 3
- Accedint a PyTorch i comprovant si hi ha disponible l'acceleració GPU/CUDA de NVIDIA
- Conclusió
Instal·lació dels controladors de la GPU de NVIDIA a Debian 12
Perquè l'acceleració GPU/CUDA de PyTorch NVIDIA funcioni, cal instal·leu els controladors de la GPU de NVIDIA a Debian 12 . Si necessiteu ajuda per instal·lar els controladors de la GPU de NVIDIA al vostre sistema Debian 12, llegiu aquest article .
Instal·lació de NVIDIA CUDA a Debian 12
Perquè l'acceleració GPU/CUDA de PyTorch NVIDIA funcioni a Debian 12, cal install NVIDIA CUDA on Debian 12 . Si necessiteu ajuda per instal·lar NVIDIA CUDA al vostre sistema Debian 12, llegiu aquest article .
Instal·lació de Python 3 PIP i Python 3 Virtual Environment (venv) a Debian 12
Per instal·lar PyTorch a Debian 12, heu de tenir instal·lats el PIP de Python 3 i l'entorn virtual de Python (venv).
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 PyTorch
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 PyTorch al directori '/opt/pytorch', executeu l'ordre següent:
$ sudo python3 -m venv / optar / pitorxaActualització de Python 3 PIP a la darrera versió a l'entorn virtual de Python 3 PyTorch
Per actualitzar Python 3 PIP a la darrera versió a l'entorn virtual Python 3 '/opt/pytorch', executeu l'ordre següent:
$ sudo / optar / pitorxa / paperera / pip3 instal·lar --actualització pip 
Instal·lació de PyTorch amb NVIDIA GPU/CUDA Acceleration Support a Debian 12
Perquè l'acceleració PyTorch NVIDIA GPU/CUDA funcioni, heu d'instal·lar la versió correcta de PyTorch que admeti la versió del controlador NVIDIA CUDA que heu instal·lat al vostre sistema Debian 12. En el moment d'escriure aquest article, PyTorch és compatible amb les versions del controlador NVIDIA CUDA 11.8 i 12.1. Per obtenir informació actualitzada sobre les versions del controlador NVIDIA CUDA que admet PyTorch, consulteu el lloc web oficial de PyTorch .
Per comprovar la versió del controlador NVIDIA CUDA que heu instal·lat al vostre sistema Debian 12, executeu l'ordre següent. Com podeu veure, tenim instal·lada la versió 11.8 de NVIDIA CUDA al nostre sistema Debian 12.
$ nvcc --versió 
Per instal·lar PyTorch amb compatibilitat amb NVIDIA CUDA 11.8 a l'entorn virtual PyTorch Python 3, executeu l'ordre següent:
$ sudo / optar / pitorxa / paperera / pip3 instal·lar torch torchvision torchaudio --index-url https: // download.pytorch.org / whl / amb 118Per instal·lar PyTorch amb compatibilitat amb NVIDIA CUDA 12.1 a l'entorn virtual PyTorch Python 3, executeu l'ordre següent:
$ sudo / optar / pitorxa / paperera / pip3 instal·lar torch torchvision torchaudioPyTorch s'està instal·lant a l'entorn virtual PyTorch Python 3. Es triga una estona a completar-se.
En aquest punt, PyTorch s'hauria d'instal·lar a l'entorn virtual PyTorch Python 3
Activació de l'entorn virtual de PyTorch Python 3
Per activar l'entorn virtual PyTorch Python '/opt/pytorch', executeu l'ordre següent:
$ . / optar / pitorxa / paperera / activarL'entorn virtual PyTorch Python 3 hauria d'estar activat.
Accedint a PyTorch i comprovant si hi ha disponible l'acceleració GPU/CUDA de NVIDIA
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 PyTorch amb la següent línia de codi:
$ torxa d'importació 
Per comprovar la versió de PyTorch que heu instal·lat, executeu la següent línia de codi. Com podeu veure, estem executant PyTorch 2.1.0 amb suport d'acceleració NVIDIA CUDA 11.8 (cu118).
$ torxa.__versió__ 
Per comprovar si PyTorch és capaç d'utilitzar la vostra GPU NVIDIA per a l'acceleració de NVIDIA CUDA, també podeu executar la següent línia de codi. Si hi ha suport de NVIDIA CUDA disponible, s'imprimirà 'True'.
$ torch.cuda.is_available ( )Si teniu instal·lades diverses GPU al vostre ordinador, podeu comprovar el nombre de GPU que PyTorch pot utilitzar amb la següent línia de codi. Com podeu veure, tenim la GPU NVIDIA (RTX 4070) instal·lada al nostre sistema Debian 12.
$ torch.cuda.device_count ( )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 Python 3 PIP i l'entorn virtual Python 3 (venv) a Debian 12. També us vam mostrar com crear un entorn virtual Python 3 per a PyTorch a Debian 12 i com instal·lar PyTorch amb NVIDIA CUDA Suport d'acceleració 11.8 i 12.1 també a Debian 12. Finalment, us vam mostrar com activar l'entorn virtual PyTorch Python i accedir a PyTorch a Debian 12.