Najnovejša različica TensorFlow izvorno ne podpira pospeševanja NVIDIA CUDA/CuDNN v operacijskih sistemih Windows 10/11. Torej, če želite nastaviti najnovejšo različico razvojnega okolja TensorFlow s pospeškom NVIDIA CUDA/cuDNN v sistemu Windows 10/11, morate to storiti prek podsistema Windows za Linux (WSL).
V tem članku vam bomo pokazali, kako namestiti WSL v Windows 10/11 in dostopati do njega. Pokazali vam bomo tudi, kako namestite najnovejšo različico TensorFlow s podporo za pospeševanje NVIDIA CUDA/cuDNN na sistemu Ubuntu WSL v sistemu Windows 10/11.
Tema vsebine:
- Namestitev gonilnikov NVIDIA GPU v Windows 10/111
- Namestitev NVIDIA CUDA in cuDNN v Windows 10/11
- Namestitev WSL v Windows 10/11
- Dostop do terminala WSL Ubuntu Linux v sistemu Windows 10/11
- Preverjanje, ali sistem Ubuntu WSL lahko dostopa do NVIDIA GPE sistema Windows 10/11
- Namestitev Python 3 PIP na sistem Ubuntu WSL
- Nadgradnja Python 3 PIP na sistem Ubuntu WSL
- Namestitev TensorFlow s podporo za pospeševanje NVIDIA CUDA/cuDNN v sistemu Ubuntu WSL
- Preverjanje, ali pospeševanje TensorFlow CUDA deluje v sistemu Ubuntu WSL
- Dostop do sistema Ubuntu WSL s kodo Visual Studio za razvoj TensorFlow
- Zaključek
Namestitev gonilnikov NVIDIA GPU v Windows 10/11
Če želite, da TensorFlow dostopa do GPE NVIDIA za pospeševanje CUDA/cuDNN v sistemu Ubuntu WSL v sistemu Windows 10/11, morate imeti v računalniku nameščen grafični procesor NVIDIA in v sistemu Windows 10/11 namestiti gonilnik GPE NVIDIA. Če imate v računalniku nameščen grafični procesor NVIDIA in potrebujete pomoč pri namestitvi gonilnika grafične kartice NVIDIA v sistemu Windows 10/11, preberi ta članek .
Namestitev NVIDIA CUDA in cuDNN v Windows 10/11
Ko namestite gonilnike NVIDIA GPE v sistem Windows 10/11, morate namestiti NVIDIA CUDA in NVIDIA cuDNN za pospeševanje TensorFlow CUDA/cuDNN, da bo delovalo v sistemu Ubuntu WSL.
Če potrebujete pomoč pri namestitvi NVIDIA CUDA v operacijski sistem Windows 10/11, preberi ta članek .
Če potrebujete pomoč pri namestitvi NVIDIA cuDNN v operacijski sistem Windows 10/11, preberite ta članek.
Namestitev WSL v Windows 10/11
Če želite namestiti WSL v Windows 10/11, odprite aplikacijo Terminal in zaženite naslednji ukaz:
$ wsl – namestitev
Kliknite »Da«.
WSL se namešča. Za dokončanje traja nekaj časa.
Ko vidite naslednji poziv, kliknite »Da«.
Namestitev se mora nadaljevati.
Namešča se operacijski sistem Ubuntu Linux. Za dokončanje traja nekaj časa.
OPOMBA: Ubuntu je privzeti operacijski sistem Windows WSL.
Na tej točki bi moral biti sistem Ubuntu Linux WSL nameščen na vašem računalniku z operacijskim sistemom Windows 10/11.
Če želite spremembe stopiti v veljavo, znova zaženite računalnik.
Ko se vaš računalnik zažene, bi se moralo prikazati terminalsko okno, ki vas prosi, da nastavite svojega prvega uporabnika Ubuntu.
Vnesite ime za uporabnika sistema Ubuntu WSL in pritisnite < Vnesite >.
Vnesite prijavno geslo za novega uporabnika in pritisnite < Vnesite >.
Znova vnesite geslo za prijavo in pritisnite < Vnesite >.
Ustvariti je treba nov uporabniški račun za sistem Ubuntu WSL in Ubuntu mora biti pripravljen za uporabo.
Dostop do terminala WSL Ubuntu Linux v sistemu Windows 10/11
Za dostop do terminala sistema Ubuntu Linux WSL, odprite aplikacijo Terminal v sistemu Windows 10/11 in kliknite na > Ubuntu .
Odprite terminal sistema Ubuntu Linux WSL.
Preverjanje, ali sistem Ubuntu WSL lahko dostopa do NVIDIA GPE sistema Windows 10/11
Če želite preveriti, ali lahko sistem Ubuntu WSL dostopa do GPU NVIDIA vašega računalnika z operacijskim sistemom Windows 10/11, zaženite naslednji ukaz s terminala sistema Ubuntu WSL:
$nvidia-smi
Če lahko sistem Ubuntu WSL dostopa do GPE NVIDIA vašega računalnika z operacijskim sistemom Windows 10/11, boste videli informacije o uporabi vašega GPE NVIDIA, kot je prikazano na naslednjem posnetku zaslona:
Namestitev Python 3 PIP na sistem Ubuntu WSL
Če želite namestiti TensorFlow v sistem Ubuntu WSL, morate imeti nameščen Python 3 PIP v sistemu Ubuntu WSL. Python 3 PIP lahko namestite na sistem Ubuntu WSL iz uradnega repozitorija paketov Ubuntu.
Najprej posodobite predpomnilnik baze podatkov paketa APT z naslednjim ukazom:
$ sudo apt posodobitev
Če želite namestiti Python 3 PIP v sistem Ubuntu WSL, zaženite naslednji ukaz:
$ sudo apt namestite python3-pip
Za potrditev namestitve pritisnite “Y” in nato pritisnite < Vnesite >.
Python 3 PIP se namešča na sistem Ubuntu WSL. Za dokončanje traja nekaj časa.
Na tej točki bi moral biti Python 3 PIP nameščen v sistemu Ubuntu WSL.
Če želite preveriti, ali je Python 3 PIP dostopen v sistemu Ubuntu WSL, zaženite naslednji ukaz:
$ pip – različica
Kot lahko vidite, imamo Python 3 PIP 22.0.2 nameščen na našem sistemu Ubuntu WSL.
Nadgradnja Python 3 PIP na sistem Ubuntu WSL
Če želite namestiti najnovejšo različico TensorFlow, morate imeti v sistemu Ubuntu WSL nameščeno najnovejšo različico Python 3 PIP.
Če želite namestiti najnovejšo različico TensorFlow, morate imeti v sistemu Ubuntu WSL nameščeno najnovejšo različico Python 3 PIP.
$ pip namestitev – nadgradnja pip
Python PIP je treba posodobiti na najnovejšo različico (različica 23.2.1 v času tega pisanja).
$ pip – različica
Namestitev TensorFlow s podporo za pospeševanje NVIDIA CUDA/cuDNN v sistemu Ubuntu WSL
Če želite namestiti TensorFlow s podporo za pospeševanje NVIDIA CUDA/cuDNN v sistemu Ubuntu WSL vašega Windows 10/11, zaženite naslednji ukaz:
$ pip namestite tensorflow[in-cuda]
TensorFlow s podporo za NVIDIA CUDA/cuDNN in zahtevanimi odvisnostmi se prenaša in namešča. Za dokončanje traja nekaj časa.
Na tej točki bi moral biti TensorFlow s podporo za NVIDIA CUDA/cuDNN nameščen v sistemu Ubuntu WSL.
Preverjanje, ali pospeševanje TensorFlow CUDA deluje v sistemu Ubuntu WSL
Če želite preveriti, ali pospeševanje TensorFlow CUDA deluje v sistemu Ubuntu WSL, odprite interaktivno lupino/tolmač Python 3 z naslednjim ukazom:
$ python3
Če želite uvoziti TensorFlow, zaženite to vrstico kode:
$ import tensorflow kot tf
Če želite preveriti, ali je bil TensorFlow pravilno uvožen, natisnite številko različice TensorFlow z naslednjo vrstico kode:
$ tf.__različica__
Kot lahko vidite, imamo TensorFlow 2.14.0 nameščen na našem sistemu Ubuntu WSL.
Če želite preveriti, ali je vaš NVIDIA GPE na voljo za pospeševanje TensorFlow CUDA, zaženite naslednjo vrstico kode:
$ tf.config.list_physical_devices('GPU')
Kot lahko vidite, je naprava GPE na voljo za TensorFlow. Torej lahko TensorFlow uporabi grafični procesor NVIDIA vašega računalnika za pospeševanje CUDA.
Za izhod iz interaktivne lupine/tolmača Python 3 zaženite naslednjo vrstico kode:
$ quit()
Dostop do sistema Ubuntu WSL s kodo Visual Studio za razvoj TensorFlow
Visual Studio Code je odličen urejevalnik kode za razvoj TensorFlow. Če želite dostopati do sistema Ubuntu WSL s kodo Visual Studio za razvoj TensorFlow in pri tem potrebujete pomoč, preberite ta članek.
Zaključek
V tem članku smo vam pokazali, kako namestiti Ubuntu Linux prek WSL v Windows 10/11. Pokazali smo vam tudi, kako dostopati do terminala sistema Ubuntu WSL v sistemu Windows 10/11 in kako namestiti najnovejšo različico TensorFlow s podporo za pospeševanje NVIDIA CUDA/cuDNN tudi v sistemu Ubuntu WSL.