TensorFlown uusin versio ei tue NVIDIA CUDA/CuDNN -kiihdytystä Windows 10/11 -käyttöjärjestelmissä. Joten jos haluat määrittää uusimman version TensorFlow-kehitysympäristöstä NVIDIA CUDA/cuDNN -kiihdytyksellä Windows 10/11 -käyttöjärjestelmään, sinun on tehtävä se Windows Subsystem for Linuxin (WSL) kautta.
Tässä artikkelissa näytämme, kuinka WSL asennetaan Windows 10/11 -käyttöjärjestelmään ja käytetään sitä. Näytämme myös, kuinka voit asentaa uusimman TensorFlow-version NVIDIA CUDA/cuDNN -kiihdytyksen tuella Ubuntu WSL -järjestelmään Windows 10/11:ssä.
Sisällön aihe:
- NVIDIA GPU -ajureiden asentaminen Windows 10/111 -käyttöjärjestelmään
- NVIDIA CUDA:n ja cuDNN:n asentaminen Windows 10/11 -käyttöjärjestelmään
- WSL:n asentaminen Windows 10/11:een
- WSL Ubuntu Linux -päätteen käyttö Windows 10/11:ssä
- Tarkistaminen, voiko Ubuntu WSL -järjestelmä käyttää Windows 10/11:n NVIDIA GPU:ta
- Python 3 PIP:n asentaminen Ubuntu WSL -järjestelmään
- Python 3 PIP:n päivitys Ubuntu WSL -järjestelmässä
- TensorFlow'n asentaminen NVIDIA CUDA/cuDNN -kiihdytyksen tuella Ubuntu WSL -järjestelmään
- Tarkistetaan, toimiiko TensorFlow CUDA Acceleration Ubuntu WSL -järjestelmässä
- Ubuntu WSL -järjestelmän käyttö Visual Studio -koodilla TensorFlow-kehitystä varten
- Johtopäätös
NVIDIA GPU -ajurien asentaminen Windows 10/11 -käyttöjärjestelmään
Jotta TensorFlow voi käyttää NVIDIA GPU:ta CUDA/cuDNN-kiihdytystä varten Ubuntu WSL -järjestelmässä Windows 10/11:ssä, tietokoneellesi on oltava asennettuna NVIDIA GPU ja asennettava NVIDIA GPU -ohjain Windows 10/11 -käyttöjärjestelmään. Jos tietokoneellesi on asennettu NVIDIA GPU ja tarvitset apua NVIDIA GPU -ohjaimen asentamisessa Windows 10/11 -käyttöjärjestelmään, lue tämä artikkeli .
NVIDIA CUDA:n ja cuDNN:n asentaminen Windows 10/11 -käyttöjärjestelmään
Kun olet asentanut NVIDIA GPU -ohjaimet Windows 10/11 -järjestelmääsi, sinun on asennettava NVIDIA CUDA ja NVIDIA cuDNN TensorFlow CUDA/cuDNN -kiihdytystä varten, jotta ne toimivat Ubuntu WSL -järjestelmässä.
Jos tarvitset apua NVIDIA CUDA:n asentamisessa Windows 10/11 -käyttöjärjestelmääsi, lue tämä artikkeli .
Jos tarvitset apua NVIDIA cuDNN:n asentamisessa Windows 10/11 -käyttöjärjestelmääsi, lue tämä artikkeli.
WSL:n asentaminen Windows 10/11:een
WSL:n asentaminen Windows 10/11:een avaa Terminal-sovellus ja suorita seuraava komento:
$ wsl -asennus
Napsauta 'Kyllä'.
WSL:ää asennetaan. Valmistuminen kestää jonkin aikaa.
Kun näet seuraavan kehotteen, napsauta 'Kyllä'.
Asennusta pitäisi jatkaa.
Ubuntu Linux -käyttöjärjestelmää asennetaan. Valmistuminen kestää jonkin aikaa.
HUOMAUTUS: Ubuntu on Windows WSL:n oletuskäyttöjärjestelmä.
Tässä vaiheessa Ubuntu Linux WSL -järjestelmä tulisi asentaa Windows 10/11 -tietokoneellesi.
Jotta muutokset tulevat voimaan, käynnistä tietokone uudelleen.
Kun tietokoneesi käynnistyy, näkyviin tulee pääteikkuna, jossa sinua pyydetään määrittämään ensimmäinen Ubuntu-käyttäjäsi.
Kirjoita Ubuntu WSL -järjestelmän käyttäjän nimi ja paina < Tulla sisään >.
Kirjoita uuden käyttäjän kirjautumissalasana ja paina < Tulla sisään >.
Kirjoita kirjautumissalasana uudelleen ja paina < Tulla sisään >.
Ubuntu WSL -järjestelmään tulee luoda uusi käyttäjätili ja Ubuntun pitäisi olla käyttövalmis.
WSL Ubuntu Linux -päätteen käyttö Windows 10/11:ssä
Päästäksesi Ubuntu Linux WSL -järjestelmän päätelaitteeseen, avaa Terminal-sovellus Windows 10/11:ssä ja napsauta > Ubuntu .
Ubuntu Linux WSL -järjestelmän pääte tulee avata.
Tarkistaminen, voiko Ubuntu WSL -järjestelmä käyttää Windows 10/11:n NVIDIA GPU:ta
Voit tarkistaa, voiko Ubuntu WSL -järjestelmä käyttää Windows 10/11 -tietokoneesi NVIDIA GPU:ta suorittamalla seuraava komento Ubuntu WSL -järjestelmän päätteestä:
$nvidia-smi
Jos Ubuntu WSL -järjestelmä voi käyttää Windows 10/11 -tietokoneesi NVIDIA GPU:ta, näet NVIDIA GPU:n käyttötiedot seuraavan kuvakaappauksen mukaisesti:
Python 3 PIP:n asentaminen Ubuntu WSL -järjestelmään
TensorFlow'n asentaminen Ubuntu WSL -järjestelmään edellyttää, että Python 3 PIP on asennettu Ubuntu WSL -järjestelmään. Voit asentaa Python 3 PIP:n Ubuntu WSL -järjestelmään Ubuntun virallisesta pakettivarastosta.
Päivitä ensin APT-pakettitietokannan välimuisti seuraavalla komennolla:
$ sudo apt päivitys
Asenna Python 3 PIP Ubuntu WSL -järjestelmään suorittamalla seuraava komento:
$ sudo apt asenna python3-pip
Vahvista asennus painamalla 'Y' ja sitten < Tulla sisään >.
Python 3 PIP asennetaan Ubuntu WSL -järjestelmään. Valmistuminen kestää jonkin aikaa.
Tässä vaiheessa Python 3 PIP tulisi asentaa Ubuntu WSL -järjestelmään.
Voit tarkistaa, onko Python 3 PIP käytettävissä Ubuntu WSL -järjestelmässä, suorittamalla seuraava komento:
$ pip -versio
Kuten näet, meillä on Python 3 PIP 22.0.2 asennettuna Ubuntu WSL -järjestelmäämme.
Python 3 PIP:n päivitys Ubuntu WSL -järjestelmässä
Uusimman TensorFlow-version asentaminen edellyttää, että Ubuntu WSL -järjestelmääsi on asennettu uusin Python 3 PIP -versio.
Uusimman TensorFlow-version asentaminen edellyttää, että Ubuntu WSL -järjestelmääsi on asennettu uusin Python 3 PIP -versio.
$ pip asennus – päivitä pip
Python PIP tulee päivittää uusimpaan versioon (versio 23.2.1 tätä kirjoitettaessa).
$ pip -versio
TensorFlow'n asentaminen NVIDIA CUDA/cuDNN -kiihdytyksen tuella Ubuntu WSL -järjestelmään
Asenna TensorFlow NVIDIA CUDA/cuDNN -kiihdytyksen tuella Windows 10/11 -käyttöjärjestelmän Ubuntu WSL -järjestelmään suorittamalla seuraava komento:
$ pip install tensorflow[and-cuda]
TensorFlow, jossa on NVIDIA CUDA/cuDNN -tuki ja tarvittavat riippuvuudet, ladataan ja asennetaan. Valmistuminen kestää jonkin aikaa.
Tässä vaiheessa TensorFlow NVIDIA CUDA/cuDNN -tuella tulisi asentaa Ubuntu WSL -järjestelmään.
Tarkistetaan, toimiiko TensorFlow CUDA Acceleration Ubuntu WSL -järjestelmässä
Voit tarkistaa, toimiiko TensorFlow CUDA -kiihdytys Ubuntu WSL -järjestelmässä avaamalla Python 3:n interaktiivinen kuori/tulkki seuraavalla komennolla:
$ python3
Tuo TensorFlow suorittamalla seuraava koodirivi:
$ tuonti tensorflow as tf
Tarkistaaksesi, onko TensorFlow tuotu oikein, tulosta TensorFlown versionumero seuraavalla koodirivillä:
$ tf.__versio__
Kuten näet, meillä on TensorFlow 2.14.0 asennettuna Ubuntu WSL -järjestelmäämme.
Tarkista, onko NVIDIA GPU käytettävissä TensorFlow CUDA -kiihdytykseen, suorittamalla seuraava koodirivi:
$ tf.config.list_physical_devices('GPU')
Kuten näet, TensorFlow:lle on saatavana GPU-laite. Joten TensorFlow voi käyttää tietokoneesi NVIDIA GPU:ta CUDA-kiihdytykseen.
Poistuaksesi Python 3:n interaktiivisesta kuoresta/tulkista, suorita seuraava koodirivi:
$ lopeta()
Ubuntu WSL -järjestelmän käyttö Visual Studio -koodilla TensorFlow-kehitystä varten
Visual Studio Code on loistava koodieditori TensorFlow-kehitykseen. Jos haluat käyttää Ubuntu WSL -järjestelmää Visual Studio Code -koodilla TensorFlow-kehitystä varten ja tarvitset siinä apua, lue tämä artikkeli.
Johtopäätös
Tässä artikkelissa näytimme sinulle, kuinka Ubuntu Linux asennetaan WSL:n kautta Windows 10/11 -käyttöjärjestelmään. Näimme myös, kuinka voit käyttää Ubuntu WSL -järjestelmän päätelaitetta Windows 10/11 -käyttöjärjestelmässä ja kuinka asentaa TensorFlow'n uusin versio NVIDIA CUDA/cuDNN -kiihdytystuella myös Ubuntu WSL -järjestelmään.