[Õpetus] GPU kiirenduse lubamine Ubuntus WSL2-s NVIDIA CUDA platvormiga

[root]

1. Ülevaade

Kuigi WSL vaikeseade võimaldab teil arendada platvormideüleseid rakendusi ilma Windowsist lahkumata, annab WSL-s GPU kiirenduse lubamine kasutajatele otsese juurdepääsu riistvarale. See toetab GPU kiirendatud AI/ML koolitust ning võimalust arendada ja testida rakendusi, mis on üles ehitatud tehnoloogiatele, nagu OpenVINO, OpenGL ja CUDA, mis sihivad Ubuntut, jäädes Windowsi.

Mida õpime:

• Kuidas paigaldada WSL2-ga ühilduv Windowsi graafiline seadme draiver
• Kuidas paigaldada NVIDIA CUDA tööriistakomplekt WSL 2 jaoks Ubuntu
• Kuidas kompileerida ja käitada CUDA näidisrakendust Ubuntus WSL2-s

Mida vajame:

• Windows 10 versioon 21H2 või uuemat füüsilist masinat, mis on varustatud NVIDIA graafikakaardi ja administraatoriõigusega seadme draiverite paigaldamiseks
• Varem paigaldatud Ubuntu WSL2-le
• Linuxi käsurea utiliitide tundmine ja Ubuntu suhtlemine WSL2-s

Eeldused:
Järgmised sammud eeldavad kindlat riistvarakonfiguratsiooni. Kuigi kontseptsioonid on teiste arhitektuuride puhul põhimõtteliselt samad, nõuavad erinevad riistvarakonfiguratsioonid sobivaid graafikadraivereid ja CUDA tööriistakomplekti.

Enne edasiliikumist veenduge, et järgmised eeltingimused on täidetud:

• Füüsiline masin, millel on Windows 10 versioon 21H2 või uuem
• NVIDIA graafikakaart
• WSL 2-le paigaldatud Ubuntu 20.04 või uuem versioon
• Interneti ühendus, mis võimaldab alla laadida mõne GB andmemahtu

2. Paigaldage WSL-i jaoks sobiv Windows vGPU draiver

Virtuaalse GPU kasutamise võimaldamiseks on vaja spetsiaalseid draivereid, mis on see, kuidas Ubuntu rakendused pääsevad juurde teie GPU riistvarale, seega peate järgima seda sammu isegi siis, kui teie süsteemidraiverid on ajakohased.

Ajakohased lingid, mis vastavad teie konkreetsele GPU-le, leiate ametlikust WSL-i dokumentatsioonist. Need leiate jaotisest Install support for Linux GUI apps > Prerequisites. Selle näite jaoks laadime alla WSL-i jaoks mõeldud NVIDIA GPU draiveri.



Märkus. See on ainus seadme draiver, mille peate paigaldama. Ärge paigaldage Ubuntu ekraanidraivereid.

Pärast allalaadimist topeltklõpsake käivitataval failil ja klõpsake nuppu Yes, et lubada programmil teie arvutis muudatusi teha.



Kinnitage vaikekataloog ja laske isetekkelise väljavõtmise protsessil jätkata.




Ilmub draiveri versiooninumbriga ekraan, mis muutub kiiresti paigaldusprogrammi peamiseks aknaks. Jätkamiseks lugege läbi ja nõustuge litsentsitingimustega.




Kinnitage viisardi vaikesätted, klõpsates nuppu Next ja oodake paigaldamise lõpuni. Lisaks teil võidakse paluda arvuti taaskäivitada.




See samm lõpeb allolevale pildile sarnase ekraaniga.



3. Paigaldage Ubuntule NVIDIA CUDA

Tavaliselt on Linuxi jaoks mõeldud CUDA tööriistakomplektiga kaasas GPU seadmedraiver. WSL2 puhul on kasutatav CUDA draiver osa süsteemi paigaldatud Windowsi draiverist ja seetõttu tuleb olla ettevaatlik, et seda Linuxi draiverit ei paigaldataks, nagu eelnevalt mainitud.

Järgmised käsud paigaldavad WSL-i spetsiifilise CUDA tööriistakomplekti versiooni 11.4 Ubuntu 20.04 AMD64 arhitektuurile. Pidage meeles, et CUDA vanemad versioonid (<=10) ei toeta WSL2. Samuti pange tähele, et CUDA tööriistakomplektide pakettide paigaldamine otse Ubuntu hoidlast ("cuda", "cuda-11-0" või "cuda" -drivers) proovib paigaldada Linuxi NVIDIA graafikadraiveri, mis pole see, mida te WSL2 puhul soovite.

wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin

sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600

wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda-repo-wsl-ubuntu-11-4-local_11.4.0-1_amd64.deb

sudo dpkg -i cuda-repo-wsl-ubuntu-11-4-local_11.4.0-1_amd64.deb

sudo apt-key add /var/cuda-repo-wsl-ubuntu-11-4-local/7fa2af80.pub

sudo apt-get update

sudo apt-get -y install cuda

Kui olete lõpetanud, peaksite nägema rida väljundeid, mis lõppevad Done.:



Palju õnne! Nüüd peaks teil olema töötav CUDA paigaldus. Testime seda järgmises etapis.

4. Koosta näidistaotlus

Valitud näidisrakendus on kaasas CUDA tööriistakomplektiga. Ühe sellise ehitamine on suurepärane viis oma CUDA paigalduse testimiseks.

Rakenduse loomiseks minge kataloogi, kuhu on paigaldatud näidised ja kirjutage sudo make:

cd /usr/local/cuda/samples/1_Utilities/deviceQuery

sudo make

Kui kõik edukas, peaks välja nägema nagu alloleval ekraanipildil. Kui olete lõpetanud, käivitage rakendus:



./deviceQuery

Peaksite nägema sarnast väljundit, mis kirjeldab teie CUDA seadistuse funktsioone:



5. Nautige Ubuntut WSL-is!

See on kõik! Selles õpetuses oleme näidanud, kuidas lubada Ubuntu GPU kiirendust WSL2-s ja demonstreerinud selle funktsionaalsust NVIDIA CUDA tööriistakomplektiga, alates paigaldamisest kuni näidisrakenduse kompileerimise ja käitamiseni.