Cea mai recentă versiune de TensorFlow nu acceptă accelerarea NVIDIA CUDA/CuDNN în mod nativ pe sistemele de operare Windows 10/11. Deci, dacă doriți să configurați cea mai recentă versiune a mediului de dezvoltare TensorFlow cu accelerarea NVIDIA CUDA/cuDNN pe Windows 10/11, trebuie să faceți acest lucru prin subsistemul Windows pentru Linux (WSL).
În acest articol, vă vom arăta cum să instalați WSL pe Windows 10/11 și să îl accesați. De asemenea, vă vom arăta cum să instalați cea mai recentă versiune de TensorFlow cu suport de accelerare NVIDIA CUDA/cuDNN pe sistemul Ubuntu WSL pe Windows 10/11.
Subiect de conținut:
- Instalarea driverelor GPU NVIDIA pe Windows 10/111
- Instalarea NVIDIA CUDA și cuDNN pe Windows 10/11
- Instalarea WSL pe Windows 10/11
- Accesarea terminalului Linux Ubuntu WSL pe Windows 10/11
- Verificarea dacă sistemul Ubuntu WSL poate accesa GPU-ul NVIDIA din Windows 10/11
- Instalarea Python 3 PIP pe sistemul Ubuntu WSL
- Actualizarea Python 3 PIP pe sistemul Ubuntu WSL
- Instalarea TensorFlow cu suport pentru accelerare NVIDIA CUDA/cuDNN pe sistemul Ubuntu WSL
- Verificarea dacă accelerarea TensorFlow CUDA funcționează pe sistemul Ubuntu WSL
- Accesarea sistemului Ubuntu WSL cu codul Visual Studio pentru dezvoltarea TensorFlow
- Concluzie
Instalarea driverelor GPU NVIDIA pe Windows 10/11
Pentru ca TensorFlow să acceseze GPU-ul NVIDIA pentru accelerarea CUDA/cuDNN pe sistemul Ubuntu WSL pe Windows 10/11, trebuie să aveți un GPU NVIDIA instalat pe computer și să instalați driverul GPU NVIDIA pe Windows 10/11. Dacă aveți un GPU NVIDIA instalat pe computer și aveți nevoie de asistență pentru instalarea driverului GPU NVIDIA pe Windows 10/11, citeste acest articol .
Instalarea NVIDIA CUDA și cuDNN pe Windows 10/11
După ce ați instalat driverele GPU NVIDIA pe sistemul dvs. Windows 10/11, trebuie să instalați NVIDIA CUDA și NVIDIA cuDNN pentru accelerarea TensorFlow CUDA/cuDNN pentru a funcționa pe sistemul Ubuntu WSL.
Dacă aveți nevoie de asistență pentru instalarea NVIDIA CUDA pe sistemul dvs. de operare Windows 10/11, citeste acest articol .
Dacă aveți nevoie de asistență pentru instalarea NVIDIA cuDNN pe sistemul dvs. de operare Windows 10/11, citiți acest articol.
Instalarea WSL pe Windows 10/11
Pentru a instala WSL pe Windows 10/11, deschideți aplicația Terminal și rulați următoarea comandă:
$ wsl –instalare
Faceți clic pe „Da”.
Se instalează WSL. Durează ceva timp pentru a finaliza.
După ce vedeți următoarea solicitare, faceți clic pe „Da”.
Instalarea ar trebui să continue.
Sistemul de operare Ubuntu Linux este în curs de instalare. Durează ceva timp pentru a finaliza.
NOTĂ: Ubuntu este sistemul de operare implicit al Windows WSL.
În acest moment, sistemul Ubuntu Linux WSL ar trebui să fie instalat pe computerul tău cu Windows 10/11.
Pentru ca modificările să aibă efect, reporniți computerul.
Odată ce computerul pornește, ar trebui să se afișeze o fereastră Terminal, care vă solicită să configurați primul utilizator Ubuntu.
Introduceți un nume pentru utilizatorul sistemului Ubuntu WSL și apăsați < introduce >.
Introduceți o parolă de conectare pentru noul utilizator și apăsați < introduce >.
Reintroduceți parola de conectare și apăsați < introduce >.
Ar trebui creat un nou cont de utilizator pentru sistemul Ubuntu WSL și Ubuntu ar trebui să fie gata de utilizare.
Accesarea terminalului Linux Ubuntu WSL pe Windows 10/11
Pentru a accesa terminalul sistemului Ubuntu Linux WSL, deschideți o aplicație Terminal pe Windows 10/11 și faceți clic pe > Ubuntu .
Terminalul sistemului Ubuntu Linux WSL ar trebui deschis.
Verificarea dacă sistemul Ubuntu WSL poate accesa GPU-ul NVIDIA din Windows 10/11
Pentru a verifica dacă sistemul Ubuntu WSL poate accesa GPU-ul NVIDIA al computerului cu Windows 10/11, rulați următoarea comandă de la terminalul sistemului Ubuntu WSL:
$nvidia-smi
Dacă sistemul Ubuntu WSL poate accesa GPU-ul NVIDIA al computerului cu Windows 10/11, veți vedea informațiile de utilizare ale GPU-ului dvs. NVIDIA, așa cum se arată în următoarea captură de ecran:
Instalarea Python 3 PIP pe sistemul Ubuntu WSL
Pentru a instala TensorFlow pe sistemul Ubuntu WSL, trebuie să aveți instalat Python 3 PIP pe sistemul Ubuntu WSL. Puteți instala Python 3 PIP pe sistemul Ubuntu WSL din depozitul oficial de pachete al Ubuntu.
Mai întâi, actualizați memoria cache a bazei de date a pachetului APT cu următoarea comandă:
$ sudo apt update
Pentru a instala Python 3 PIP pe sistemul Ubuntu WSL, rulați următoarea comandă:
$ sudo apt install python3-pip
Pentru a confirma instalarea, apăsați „Y” și apoi apăsați < introduce >.
Python 3 PIP este instalat pe sistemul Ubuntu WSL. Durează ceva timp pentru a finaliza.
În acest moment, Python 3 PIP ar trebui să fie instalat pe sistemul Ubuntu WSL.
Pentru a verifica dacă Python 3 PIP este accesibil pe sistemul Ubuntu WSL, rulați următoarea comandă:
$ pip – versiune
După cum puteți vedea, avem instalat Python 3 PIP 22.0.2 pe sistemul nostru Ubuntu WSL.
Actualizarea Python 3 PIP pe sistemul Ubuntu WSL
Pentru a instala cea mai recentă versiune de TensorFlow, trebuie să aveți cea mai recentă versiune de Python 3 PIP instalată pe sistemul dumneavoastră Ubuntu WSL.
Pentru a instala cea mai recentă versiune de TensorFlow, trebuie să aveți cea mai recentă versiune de Python 3 PIP instalată pe sistemul dumneavoastră Ubuntu WSL.
$ pip install –upgrade pip
Python PIP ar trebui să fie actualizat la cea mai recentă versiune (versiunea 23.2.1 la momentul scrierii acestui articol).
$ pip –versiune
Instalarea TensorFlow cu suport pentru accelerare NVIDIA CUDA/cuDNN pe sistemul Ubuntu WSL
Pentru a instala TensorFlow cu suport de accelerare NVIDIA CUDA/cuDNN pe sistemul Ubuntu WSL al Windows 10/11, rulați următoarea comandă:
$ pip install tensorflow[and-cuda]
TensorFlow cu suport NVIDIA CUDA/cuDNN și dependențele necesare sunt descărcate și instalate. Durează ceva timp pentru a finaliza.
În acest moment, TensorFlow cu suport NVIDIA CUDA/cuDNN ar trebui să fie instalat pe sistemul Ubuntu WSL.
Verificarea dacă accelerarea TensorFlow CUDA funcționează pe sistemul Ubuntu WSL
Pentru a verifica dacă accelerarea TensorFlow CUDA funcționează pe sistemul Ubuntu WSL, deschideți shell-ul/interpretul interactiv Python 3 cu următoarea comandă:
$ python3
Pentru a importa TensorFlow, rulați următoarea linie de cod:
$ import tensorflow ca tf
Pentru a verifica dacă TensorFlow a fost importat corect, tipăriți numărul versiunii TensorFlow cu următoarea linie de cod:
$ tf.__versiune__
După cum puteți vedea, avem TensorFlow 2.14.0 instalat pe sistemul nostru Ubuntu WSL.
Pentru a verifica dacă GPU-ul dvs. NVIDIA este disponibil pentru accelerarea TensorFlow CUDA, rulați următoarea linie de cod:
$ tf.config.list_physical_devices(‘GPU’)
După cum puteți vedea, un dispozitiv GPU este disponibil pentru TensorFlow. Deci, TensorFlow poate folosi GPU-ul NVIDIA al computerului dvs. pentru accelerarea CUDA.
Pentru a ieși din shell-ul/interpretul interactiv Python 3, rulați următoarea linie de cod:
$ renunta()
Accesarea sistemului Ubuntu WSL cu codul Visual Studio pentru dezvoltarea TensorFlow
Visual Studio Code este un editor de cod excelent pentru dezvoltarea TensorFlow. Dacă doriți să accesați sistemul Ubuntu WSL cu Visual Studio Code pentru dezvoltarea TensorFlow și aveți nevoie de asistență în acest sens, citiți acest articol.
Concluzie
În acest articol, v-am arătat cum să instalați Ubuntu Linux prin WSL pe Windows 10/11. De asemenea, v-am arătat cum să accesați terminalul sistemului Ubuntu WSL pe Windows 10/11 și cum să instalați cea mai recentă versiune de TensorFlow cu suportul de accelerare NVIDIA CUDA/cuDNN și pe sistemul Ubuntu WSL.