Kā instalēt jaunāko TensorFlow operētājsistēmā Windows 10/11 ar NVIDIA CUDA/cuDNN paātrinājumu, izmantojot WSL

Jaunākā TensorFlow versija neatbalsta NVIDIA CUDA/CuDNN paātrinājumu sākotnēji operētājsistēmās Windows 10/11. Tātad, ja vēlaties iestatīt jaunāko TensorFlow izstrādes vides versiju ar NVIDIA CUDA/cuDNN paātrinājumu operētājsistēmā Windows 10/11, jums tas jādara, izmantojot Windows apakšsistēmu Linux (WSL).

Šajā rakstā mēs parādīsim, kā instalēt WSL operētājsistēmā Windows 10/11 un piekļūt tai. Mēs arī parādīsim, kā instalēt jaunāko TensorFlow versiju ar NVIDIA CUDA/cuDNN paātrinājuma atbalstu Ubuntu WSL sistēmā operētājsistēmā Windows 10/11.

Satura tēma:

1. NVIDIA GPU draiveru instalēšana operētājsistēmā Windows 10/111
2. NVIDIA CUDA un cuDNN instalēšana operētājsistēmā Windows 10/11
3. WSL instalēšana operētājsistēmā Windows 10/11
4. Piekļuve WSL Ubuntu Linux terminālim operētājsistēmā Windows 10/11
5. Pārbaude, vai Ubuntu WSL sistēma var piekļūt operētājsistēmas Windows 10/11 NVIDIA GPU
6. Python 3 PIP instalēšana Ubuntu WSL sistēmā
7. Python 3 PIP jaunināšana Ubuntu WSL sistēmā
8. TensorFlow instalēšana ar NVIDIA CUDA/cuDNN paātrinājuma atbalstu Ubuntu WSL sistēmā
9. Pārbaude, vai TensorFlow CUDA paātrinājums darbojas Ubuntu WSL sistēmā
10. Piekļuve Ubuntu WSL sistēmai ar Visual Studio kodu TensorFlow izstrādei
11. Secinājums

NVIDIA GPU draiveru instalēšana operētājsistēmā Windows 10/11

Lai TensorFlow varētu piekļūt NVIDIA GPU CUDA/cuDNN paātrinājumam Ubuntu WSL sistēmā operētājsistēmā Windows 10/11, datorā ir jābūt instalētam NVIDIA GPU un jāinstalē NVIDIA GPU draiveris operētājsistēmā Windows 10/11. Ja jūsu datorā ir instalēts NVIDIA GPU un jums ir nepieciešama palīdzība, instalējot NVIDIA GPU draiveri operētājsistēmā Windows 10/11, izlasi šo rakstu .

NVIDIA CUDA un cuDNN instalēšana operētājsistēmā Windows 10/11

Kad esat instalējis NVIDIA GPU draiverus savā Windows 10/11 sistēmā, jums ir jāinstalē NVIDIA CUDA un NVIDIA cuDNN TensorFlow CUDA/cuDNN paātrināšanai, lai darbotos Ubuntu WSL sistēmā.

Ja jums nepieciešama palīdzība, instalējot NVIDIA CUDA operētājsistēmā Windows 10/11, izlasi šo rakstu .

Ja jums ir nepieciešama palīdzība, instalējot NVIDIA cuDNN operētājsistēmā Windows 10/11, izlasiet šo rakstu.

WSL instalēšana operētājsistēmā Windows 10/11

Lai instalētu WSL operētājsistēmā Windows 10/11, atveriet lietotni Terminal un palaidiet šādu komandu:

$ wsl – instalēt

Noklikšķiniet uz 'Jā'.

WSL tiek instalēts. Tas aizņem kādu laiku, lai pabeigtu.

Kad redzat šādu uzvedni, noklikšķiniet uz 'Jā'.

Instalēšana jāturpina.

Tiek instalēta operētājsistēma Ubuntu Linux. Tas aizņem kādu laiku, lai pabeigtu.
PIEZĪME: Ubuntu ir Windows WSL noklusējuma operētājsistēma.

Šajā brīdī jūsu Windows 10/11 datorā ir jāinstalē Ubuntu Linux WSL sistēma.

Lai izmaiņas stātos spēkā, restartējiet datoru.

Kad dators ir sāknēts, tiek parādīts termināļa logs, kurā tiek lūgts iestatīt savu pirmo Ubuntu lietotāju.
Ievadiet Ubuntu WSL sistēmas lietotāja vārdu un nospiediet < Ievadiet >.

Ievadiet jaunā lietotāja pieteikšanās paroli un nospiediet < Ievadiet >.

Atkārtoti ierakstiet pieteikšanās paroli un nospiediet < Ievadiet >.

Jāizveido jauns lietotāja konts Ubuntu WSL sistēmai, un Ubuntu jābūt gatavam lietošanai.

Piekļuve WSL Ubuntu Linux terminālim operētājsistēmā Windows 10/11

Lai piekļūtu Ubuntu Linux WSL sistēmas terminālim, atveriet termināļa lietotni operētājsistēmā Windows 10/11 un noklikšķiniet uz > Ubuntu .

Ir jāatver Ubuntu Linux WSL sistēmas terminālis.

Pārbaude, vai Ubuntu WSL sistēma var piekļūt operētājsistēmas Windows 10/11 NVIDIA GPU

Lai pārbaudītu, vai Ubuntu WSL sistēma var piekļūt jūsu Windows 10/11 datora NVIDIA GPU, palaidiet šo komandu no Ubuntu WSL sistēmas termināļa:

$nvidia-smi

Ja Ubuntu WSL sistēma var piekļūt jūsu Windows 10/11 datora NVIDIA GPU, jūs redzēsit sava NVIDIA GPU lietojuma informāciju, kā parādīts šajā ekrānuzņēmumā:

Python 3 PIP instalēšana Ubuntu WSL sistēmā

Lai instalētu TensorFlow Ubuntu WSL sistēmā, Ubuntu WSL sistēmā ir jābūt instalētam Python 3 PIP. Python 3 PIP var instalēt Ubuntu WSL sistēmā no Ubuntu oficiālās pakotņu krātuves.

Vispirms atjauniniet APT pakotnes datu bāzes kešatmiņu ar šādu komandu:

$ sudo apt atjauninājums

Lai instalētu Python 3 PIP Ubuntu WSL sistēmā, palaidiet šo komandu:

$ sudo apt instalējiet python3-pip

Lai apstiprinātu instalēšanu, nospiediet “Y” un pēc tam nospiediet < Ievadiet >.

Python 3 PIP tiek instalēta Ubuntu WSL sistēmā. Tas aizņem kādu laiku, lai pabeigtu.

Šajā brīdī Python 3 PIP ir jāinstalē Ubuntu WSL sistēmā.

Lai pārbaudītu, vai Python 3 PIP ir pieejams Ubuntu WSL sistēmā, palaidiet šo komandu:

$ pip – versija

Kā redzat, mūsu Ubuntu WSL sistēmā ir instalēta Python 3 PIP 22.0.2.

Python 3 PIP jaunināšana Ubuntu WSL sistēmā

Lai instalētu jaunāko TensorFlow versiju, Ubuntu WSL sistēmā ir jābūt instalētai jaunākajai Python 3 PIP versijai.

Lai instalētu jaunāko TensorFlow versiju, Ubuntu WSL sistēmā ir jābūt instalētai jaunākajai Python 3 PIP versijai.

$ pip instalēšana – atjaunināšana pip

Python PIP ir jāatjaunina uz jaunāko versiju (šī rakstīšanas laikā versija 23.2.1).

$ pip – versija

TensorFlow instalēšana ar NVIDIA CUDA/cuDNN paātrinājuma atbalstu Ubuntu WSL sistēmā

Lai instalētu TensorFlow ar NVIDIA CUDA/cuDNN paātrinājuma atbalstu Windows 10/11 Ubuntu WSL sistēmā, palaidiet šo komandu:

$ pip instalēt tensorflow[and-cuda]

TensorFlow ar NVIDIA CUDA/cuDNN atbalstu un nepieciešamajām atkarībām tiek lejupielādēta un instalēta. Tas aizņem kādu laiku, lai pabeigtu.

Šajā brīdī Ubuntu WSL sistēmā ir jāinstalē TensorFlow ar NVIDIA CUDA/cuDNN atbalstu.

Pārbaude, vai TensorFlow CUDA paātrinājums darbojas Ubuntu WSL sistēmā

Lai pārbaudītu, vai TensorFlow CUDA paātrinājums darbojas Ubuntu WSL sistēmā, atveriet Python 3 interaktīvo apvalku/tulku ar šādu komandu:

$ python3

Lai importētu TensorFlow, palaidiet šādu koda rindiņu:

$ importēt tensorflow kā tf

Lai pārbaudītu, vai TensorFlow tika importēts pareizi, izdrukājiet TensorFlow versijas numuru ar šādu koda rindiņu:

$ tf.__version__

Kā redzat, mūsu Ubuntu WSL sistēmā ir instalēta TensorFlow 2.14.0.

Lai pārbaudītu, vai jūsu NVIDIA GPU ir pieejams TensorFlow CUDA paātrinājumam, palaidiet šo koda rindiņu:

$ tf.config.list_physical_devices('GPU')

Kā redzat, TensorFlow ir pieejama GPU ierīce. Tātad TensorFlow var izmantot jūsu datora NVIDIA GPU CUDA paātrināšanai.

Lai izietu no Python 3 interaktīvās čaulas/tulka, palaidiet šo koda rindiņu:

$ iziet()

Piekļuve Ubuntu WSL sistēmai ar Visual Studio kodu TensorFlow izstrādei

Visual Studio Code ir lielisks koda redaktors TensorFlow izstrādei. Ja vēlaties piekļūt Ubuntu WSL sistēmai ar Visual Studio kodu TensorFlow izstrādei un jums ir nepieciešama palīdzība, izlasiet šo rakstu.

Secinājums

Šajā rakstā mēs parādījām, kā instalēt Ubuntu Linux, izmantojot WSL operētājsistēmā Windows 10/11. Mēs arī parādījām, kā piekļūt Ubuntu WSL sistēmas terminālim operētājsistēmā Windows 10/11 un kā instalēt jaunāko TensorFlow versiju ar NVIDIA CUDA/cuDNN paātrinājuma atbalstu arī Ubuntu WSL sistēmā.