La darrera versió de TensorFlow no admet l'acceleració NVIDIA CUDA/CuDNN de manera nativa als sistemes operatius Windows 10/11. Per tant, si voleu configurar la darrera versió de l'entorn de desenvolupament TensorFlow amb l'acceleració NVIDIA CUDA/cuDNN a Windows 10/11, ho heu de fer mitjançant el subsistema de Windows per a Linux (WSL).
En aquest article, us mostrarem com instal·lar WSL a Windows 10/11 i accedir-hi. També us mostrarem com instal·lar la darrera versió de TensorFlow amb suport d'acceleració NVIDIA CUDA/cuDNN al sistema Ubuntu WSL a Windows 10/11.
Tema de continguts:
- Instal·lació dels controladors de GPU de NVIDIA a Windows 10/111
- Instal·lació de NVIDIA CUDA i cuDNN a Windows 10/11
- Instal·lació de WSL a Windows 10/11
- Accés al terminal Linux Ubuntu WSL a Windows 10/11
- Comprovar si el sistema Ubuntu WSL pot accedir a la GPU NVIDIA de Windows 10/11
- Instal·lació de Python 3 PIP al sistema Ubuntu WSL
- Actualització de Python 3 PIP al sistema Ubuntu WSL
- Instal·lació de TensorFlow amb suport d'acceleració NVIDIA CUDA/cuDNN al sistema Ubuntu WSL
- Comprovant si TensorFlow CUDA Acceleration funciona al sistema Ubuntu WSL
- Accés al sistema Ubuntu WSL amb Visual Studio Code per al desenvolupament de TensorFlow
- Conclusió
Instal·lació dels controladors de GPU de NVIDIA a Windows 10/11
Perquè TensorFlow pugui accedir a la GPU NVIDIA per a l'acceleració CUDA/cuDNN al sistema Ubuntu WSL a Windows 10/11, heu de tenir una GPU NVIDIA instal·lada a l'ordinador i instal·lar el controlador de GPU NVIDIA a Windows 10/11. Si teniu una GPU NVIDIA instal·lada a l'ordinador i necessiteu ajuda per instal·lar el controlador de la GPU NVIDIA a Windows 10/11, llegiu aquest article .
Instal·lació de NVIDIA CUDA i cuDNN a Windows 10/11
Un cop instal·lats els controladors de GPU de NVIDIA al vostre sistema Windows 10/11, heu d'instal·lar NVIDIA CUDA i NVIDIA cuDNN per a l'acceleració TensorFlow CUDA/cuDNN per treballar al sistema Ubuntu WSL.
Si necessiteu ajuda per instal·lar NVIDIA CUDA al vostre sistema operatiu Windows 10/11, llegiu aquest article .
Si necessiteu ajuda per instal·lar NVIDIA cuDNN al vostre sistema operatiu Windows 10/11, llegiu aquest article.
Instal·lació de WSL a Windows 10/11
Per instal·lar WSL a Windows 10/11, obriu l'aplicació Terminal i executeu l'ordre següent:
$ wsl –instal·lar
Feu clic a 'Sí'.
S'està instal·lant WSL. Es triga una estona a completar-se.
Quan vegis el següent missatge, fes clic a 'Sí'.
La instal·lació hauria de continuar.
S'està instal·lant el sistema operatiu Ubuntu Linux. Es triga una estona a completar-se.
NOTA: Ubuntu és el sistema operatiu predeterminat de Windows WSL.
En aquest punt, el sistema Ubuntu Linux WSL s'hauria d'instal·lar al vostre ordinador Windows 10/11.
Perquè els canvis tinguin efecte, reinicieu l'ordinador.
Un cop arrenqui l'ordinador, s'hauria de mostrar una finestra de Terminal que us demanarà que configureu el vostre primer usuari d'Ubuntu.
Escriviu un nom per a l'usuari del sistema Ubuntu WSL i premeu < Entra >.
Introduïu una contrasenya d'inici de sessió per al nou usuari i premeu < Entra >.
Torneu a escriure la contrasenya d'inici de sessió i premeu < Entra >.
S'hauria de crear un compte d'usuari nou per al sistema Ubuntu WSL i Ubuntu hauria d'estar llest per utilitzar-lo.
Accés al terminal Linux Ubuntu WSL a Windows 10/11
Per accedir al terminal del sistema Ubuntu Linux WSL, Obriu una aplicació Terminal a Windows 10/11 i feu clic a > Ubuntu .
S'hauria d'obrir el terminal del sistema Ubuntu Linux WSL.
Comprovar si el sistema Ubuntu WSL pot accedir a la GPU NVIDIA de Windows 10/11
Per comprovar si el sistema Ubuntu WSL pot accedir a la GPU NVIDIA del vostre ordinador Windows 10/11, executeu l'ordre següent des del terminal del sistema Ubuntu WSL:
$nvidia-smi
Si el sistema Ubuntu WSL pot accedir a la GPU NVIDIA del vostre ordinador Windows 10/11, veureu la informació d'ús de la vostra GPU NVIDIA tal com es mostra a la captura de pantalla següent:
Instal·lació de Python 3 PIP al sistema Ubuntu WSL
Per instal·lar TensorFlow al sistema Ubuntu WSL, heu de tenir instal·lat Python 3 PIP al sistema Ubuntu WSL. Podeu instal·lar Python 3 PIP al sistema Ubuntu WSL des del dipòsit oficial de paquets d'Ubuntu.
Primer, actualitzeu la memòria cau de la base de dades del paquet APT amb l'ordre següent:
$ sudo apt update
Per instal·lar Python 3 PIP al sistema Ubuntu WSL, executeu l'ordre següent:
$ sudo apt install python3-pip
Per confirmar la instal·lació, premeu 'Y' i després premeu < Entra >.
Python 3 PIP s'està instal·lant al sistema Ubuntu WSL. Es triga una estona a completar-se.
En aquest punt, Python 3 PIP s'hauria d'instal·lar al sistema Ubuntu WSL.
Per comprovar si Python 3 PIP és accessible al sistema Ubuntu WSL, executeu l'ordre següent:
$ pip –versió
Com podeu veure, tenim instal·lat Python 3 PIP 22.0.2 al nostre sistema Ubuntu WSL.
Actualització de Python 3 PIP al sistema Ubuntu WSL
Per instal·lar la darrera versió de TensorFlow, heu de tenir la darrera versió de Python 3 PIP instal·lada al vostre sistema Ubuntu WSL.
Per instal·lar la darrera versió de TensorFlow, heu de tenir la darrera versió de Python 3 PIP instal·lada al vostre sistema Ubuntu WSL.
$ pip instal·lació –actualització pip
Python PIP s'hauria d'actualitzar a la darrera versió (versió 23.2.1 en el moment d'escriure aquest article).
$ pip –versió
Instal·lació de TensorFlow amb suport d'acceleració NVIDIA CUDA/cuDNN al sistema Ubuntu WSL
Per instal·lar TensorFlow amb suport d'acceleració NVIDIA CUDA/cuDNN al sistema Ubuntu WSL del vostre Windows 10/11, executeu l'ordre següent:
$ pip instal·lar tensorflow[i-cuda]
S'estan baixant i instal·lant TensorFlow amb suport NVIDIA CUDA/cuDNN i les dependències necessàries. Es triga una estona a completar-se.
En aquest punt, TensorFlow amb suport NVIDIA CUDA/cuDNN s'hauria d'instal·lar al sistema Ubuntu WSL.
Comprovant si TensorFlow CUDA Acceleration funciona al sistema Ubuntu WSL
Per comprovar si l'acceleració de TensorFlow CUDA funciona al sistema Ubuntu WSL, obriu l'intèrpret/intèrpret interactiu de Python 3 amb l'ordre següent:
$ python3
Per importar TensorFlow, executeu la línia de codi següent:
$ importar tensorflow com a tf
Per comprovar si TensorFlow s'ha importat correctament, imprimiu el número de versió de TensorFlow amb la següent línia de codi:
$ tf.__versió__
Com podeu veure, tenim TensorFlow 2.14.0 instal·lat al nostre sistema Ubuntu WSL.
Per comprovar si la vostra GPU NVIDIA està disponible per a l'acceleració TensorFlow CUDA, executeu la següent línia de codi:
$ tf.config.list_physical_devices(‘GPU’)
Com podeu veure, un dispositiu GPU està disponible per a TensorFlow. Per tant, TensorFlow pot utilitzar la GPU NVIDIA del vostre ordinador per accelerar CUDA.
Per sortir de l'intèrpret/intèrpret interactiu de Python 3, executeu la següent línia de codi:
$ sortir()
Accés al sistema Ubuntu WSL amb Visual Studio Code per al desenvolupament de TensorFlow
Visual Studio Code és un gran editor de codi per al desenvolupament de TensorFlow. Si voleu accedir al sistema Ubuntu WSL amb Visual Studio Code per al desenvolupament de TensorFlow i necessiteu ajuda, llegiu aquest article.
Conclusió
En aquest article, us vam mostrar com instal·lar Ubuntu Linux mitjançant WSL a Windows 10/11. També us vam mostrar com accedir al terminal del sistema Ubuntu WSL a Windows 10/11 i com instal·lar la darrera versió de TensorFlow amb el suport d'acceleració NVIDIA CUDA/cuDNN també al sistema Ubuntu WSL.