Naujausia „TensorFlow“ versija „Windows 10/11“ operacinėse sistemose nepalaiko NVIDIA CUDA / CuDNN spartinimo. Taigi, jei norite nustatyti naujausią TensorFlow kūrimo aplinkos versiją su NVIDIA CUDA / cuDNN spartinimu Windows 10/11, turite tai padaryti naudodami Windows posistemę, skirtą Linux (WSL).
Šiame straipsnyje parodysime, kaip įdiegti WSL sistemoje „Windows 10/11“ ir ją pasiekti. Taip pat parodysime, kaip įdiegti naujausią TensorFlow versiją su NVIDIA CUDA / cuDNN spartinimo palaikymu Ubuntu WSL sistemoje Windows 10/11.
Turinio tema:
- NVIDIA GPU tvarkyklių diegimas sistemoje „Windows 10/111“.
- NVIDIA CUDA ir cuDNN diegimas sistemoje „Windows 10/11“.
- WSL diegimas sistemoje „Windows 10/11“.
- Prieiga prie WSL Ubuntu Linux terminalo sistemoje Windows 10/11
- Tikrinama, ar Ubuntu WSL sistema gali pasiekti „Windows 10/11“ NVIDIA GPU
- Python 3 PIP diegimas Ubuntu WSL sistemoje
- Python 3 PIP atnaujinimas Ubuntu WSL sistemoje
- „TensorFlow“ diegimas su NVIDIA CUDA / cuDNN spartinimo palaikymu Ubuntu WSL sistemoje
- Tikrinama, ar „TensorFlow CUDA Acceleration“ veikia Ubuntu WSL sistemoje
- Prieiga prie Ubuntu WSL sistemos su Visual Studio kodu, skirtu TensorFlow plėtrai
- Išvada
NVIDIA GPU tvarkyklių diegimas sistemoje „Windows 10/11“.
Kad „TensorFlow“ galėtų pasiekti NVIDIA GPU, skirtą CUDA / cuDNN spartinimui Ubuntu WSL sistemoje „Windows 10/11“, kompiuteryje turi būti įdiegtas NVIDIA GPU ir „Windows 10/11“ sistemoje įdiegta NVIDIA GPU tvarkyklė. Jei kompiuteryje įdiegtas NVIDIA GPU ir jums reikia pagalbos diegiant NVIDIA GPU tvarkyklę sistemoje Windows 10/11, perskaitykite šį straipsnį .
NVIDIA CUDA ir cuDNN diegimas sistemoje „Windows 10/11“.
Įdiegę NVIDIA GPU tvarkykles savo Windows 10/11 sistemoje, turite įdiegti NVIDIA CUDA ir NVIDIA cuDNN, kad būtų galima pagreitinti TensorFlow CUDA / cuDNN, kad veiktų Ubuntu WSL sistemoje.
Jei jums reikia pagalbos diegiant NVIDIA CUDA operacinėje sistemoje „Windows 10/11“, perskaitykite šį straipsnį .
Jei jums reikia pagalbos diegiant NVIDIA cuDNN operacinėje sistemoje „Windows 10/11“, perskaitykite šį straipsnį.
WSL diegimas sistemoje „Windows 10/11“.
Norėdami įdiegti WSL sistemoje „Windows 10/11“, atidarykite terminalo programą ir paleiskite šią komandą:
$ wsl – įdiegti
Spustelėkite „Taip“.
WSL diegiamas. Užbaigti užtrunka šiek tiek laiko.
Kai pamatysite šį raginimą, spustelėkite „Taip“.
Diegimas turėtų tęstis.
Diegiama Ubuntu Linux operacinė sistema. Užbaigti užtrunka šiek tiek laiko.
PASTABA: Ubuntu yra numatytoji Windows WSL operacinė sistema.
Šiuo metu „Windows 10/11“ kompiuteryje turėtų būti įdiegta „Ubuntu Linux WSL“ sistema.
Kad pakeitimai įsigaliotų, iš naujo paleiskite kompiuterį.
Kai kompiuteris paleidžiamas, turėtų būti rodomas terminalo langas, kuriame prašoma nustatyti pirmąjį Ubuntu vartotoją.
Įveskite Ubuntu WSL sistemos vartotojo vardą ir paspauskite < Įeikite >.
Įveskite naujo vartotojo prisijungimo slaptažodį ir paspauskite < Įeikite >.
Dar kartą įveskite prisijungimo slaptažodį ir paspauskite < Įeikite >.
Ubuntu WSL sistemai turėtų būti sukurta nauja vartotojo paskyra ir Ubuntu turėtų būti paruoštas naudoti.
Prieiga prie WSL Ubuntu Linux terminalo sistemoje Windows 10/11
Norėdami pasiekti Ubuntu Linux WSL sistemos terminalą, atidarykite terminalo programą „Windows 10/11“. ir spustelėkite > Ubuntu .
Turėtų būti atidarytas Ubuntu Linux WSL sistemos terminalas.
Tikrinama, ar Ubuntu WSL sistema gali pasiekti „Windows 10/11“ NVIDIA GPU
Norėdami patikrinti, ar Ubuntu WSL sistema gali pasiekti jūsų Windows 10/11 kompiuterio NVIDIA GPU, paleiskite šią komandą iš Ubuntu WSL sistemos terminalo:
$nvidia-smi
Jei Ubuntu WSL sistema gali pasiekti jūsų Windows 10/11 kompiuterio NVIDIA GPU, matysite savo NVIDIA GPU naudojimo informaciją, kaip parodyta šioje ekrano kopijoje:
Python 3 PIP diegimas Ubuntu WSL sistemoje
Norėdami įdiegti TensorFlow Ubuntu WSL sistemoje, Ubuntu WSL sistemoje turite įdiegti Python 3 PIP. Python 3 PIP galite įdiegti Ubuntu WSL sistemoje iš oficialios Ubuntu paketų saugyklos.
Pirmiausia atnaujinkite APT paketo duomenų bazės talpyklą naudodami šią komandą:
$ sudo apt atnaujinimas
Norėdami įdiegti Python 3 PIP Ubuntu WSL sistemoje, paleiskite šią komandą:
$ sudo apt įdiegti python3-pip
Norėdami patvirtinti diegimą, paspauskite „Y“, tada paspauskite < Įeikite >.
Python 3 PIP diegiamas Ubuntu WSL sistemoje. Užbaigti užtrunka šiek tiek laiko.
Šiuo metu Ubuntu WSL sistemoje turėtų būti įdiegtas Python 3 PIP.
Norėdami patikrinti, ar Python 3 PIP galima pasiekti Ubuntu WSL sistemoje, paleiskite šią komandą:
$ pip – versija
Kaip matote, mūsų Ubuntu WSL sistemoje įdiegtas Python 3 PIP 22.0.2.
Python 3 PIP atnaujinimas Ubuntu WSL sistemoje
Norėdami įdiegti naujausią TensorFlow versiją, Ubuntu WSL sistemoje turite turėti naujausią Python 3 PIP versiją.
Norėdami įdiegti naujausią TensorFlow versiją, Ubuntu WSL sistemoje turite turėti naujausią Python 3 PIP versiją.
$ pip įdiegti – atnaujinti pip
Python PIP turėtų būti atnaujintas į naujausią versiją (šio rašymo metu versija 23.2.1).
$ pip – versija
„TensorFlow“ diegimas su NVIDIA CUDA / cuDNN spartinimo palaikymu Ubuntu WSL sistemoje
Norėdami įdiegti „TensorFlow“ su NVIDIA CUDA / cuDNN spartinimo palaikymu „Windows 10/11“ Ubuntu WSL sistemoje, paleiskite šią komandą:
$ pip įdiegti tensorflow[and-cuda]
Atsisiunčiama ir įdiegiama „TensorFlow“ su NVIDIA CUDA/cuDNN palaikymu ir reikiamomis priklausomybėmis. Užbaigti užtrunka šiek tiek laiko.
Šiuo metu „TensorFlow“ su NVIDIA CUDA/cuDNN palaikymu turėtų būti įdiegta „Ubuntu WSL“ sistemoje.
Tikrinama, ar „TensorFlow CUDA Acceleration“ veikia Ubuntu WSL sistemoje
Norėdami patikrinti, ar TensorFlow CUDA pagreitis veikia Ubuntu WSL sistemoje, atidarykite Python 3 interaktyvų apvalkalą / interpretatorių naudodami šią komandą:
$ python3
Norėdami importuoti TensorFlow, paleiskite šią kodo eilutę:
$ importuoti tensorflow kaip tf
Norėdami patikrinti, ar TensorFlow buvo importuotas teisingai, atspausdinkite TensorFlow versijos numerį naudodami šią kodo eilutę:
$ tf.__version__
Kaip matote, mūsų Ubuntu WSL sistemoje įdiegta TensorFlow 2.14.0.
Norėdami patikrinti, ar jūsų NVIDIA GPU galima naudoti TensorFlow CUDA pagreitinimui, paleiskite šią kodo eilutę:
$ tf.config.list_physical_devices('GPU')
Kaip matote, „TensorFlow“ yra GPU įrenginys. Taigi, „TensorFlow“ gali naudoti jūsų kompiuterio NVIDIA GPU CUDA pagreitinimui.
Norėdami išeiti iš Python 3 interaktyvaus apvalkalo / vertėjo, paleiskite šią kodo eilutę:
$ išeiti ()
Prieiga prie Ubuntu WSL sistemos su Visual Studio kodu, skirtu TensorFlow plėtrai
„Visual Studio Code“ yra puikus „TensorFlow“ kūrimo kodo rengyklė. Jei norite pasiekti Ubuntu WSL sistemą su „Visual Studio Code“, skirtu „TensorFlow“ plėtrai, ir jums reikia bet kokios pagalbos, perskaitykite šį straipsnį.
Išvada
Šiame straipsnyje mes jums parodėme, kaip įdiegti „Ubuntu Linux“ per WSL „Windows 10/11“. Taip pat parodėme, kaip pasiekti Ubuntu WSL sistemos terminalą sistemoje Windows 10/11 ir kaip įdiegti naujausią TensorFlow versiją su NVIDIA CUDA/cuDNN pagreičio palaikymu Ubuntu WSL sistemoje.