PyTorch može ubrzati AI/ML aplikacije pomoću NVIDIA GPU-a putem NVIDIA CUDA biblioteke izvorno kao TensorFlow.
U ovom ćemo vam članku pokazati kako instalirati PyTorch s podrškom za NVIDIA GPU/CUDA ubrzanje na Debian 12 “Bookworm”.
Tema sadržaja:
- Instaliranje NVIDIA GPU upravljačkih programa na Debian 12
- Instaliranje NVIDIA CUDA na Debian 12
- Instaliranje Python 3 PIP i Python 3 Virtual Environment (venv) na Debian 12
- Stvaranje Python 3 virtualnog okruženja za PyTorch
- Nadogradnja Python 3 PIP-a na najnoviju verziju na Python 3 PyTorch virtualnom okruženju
- Instaliranje PyTorcha s NVIDIA GPU/CUDA podrškom za ubrzanje na Debian 12
- Aktiviranje virtualnog okruženja PyTorch Python 3
- Pristup PyTorchu i provjera je li dostupno NVIDIA GPU/CUDA ubrzanje
- Zaključak
Instaliranje NVIDIA GPU upravljačkih programa na Debian 12
Da bi PyTorch NVIDIA GPU/CUDA ubrzanje radilo, morate instalirajte NVIDIA GPU upravljačke programe na Debian 12 . Ako trebate bilo kakvu pomoć pri instaliranju NVIDIA GPU upravljačkih programa na vašem Debian 12 sustavu, pročitajte ovaj članak .
Instaliranje NVIDIA CUDA na Debian 12
Da bi PyTorch NVIDIA GPU/CUDA akceleracija radila na Debianu 12, morate instalirajte NVIDIA CUDA na Debian 12 . Ako trebate bilo kakvu pomoć pri instaliranju NVIDIA CUDA na vaš Debian 12 sustav, pročitajte ovaj članak .
Instaliranje Python 3 PIP i Python 3 Virtual Environment (venv) na Debian 12
Da biste instalirali PyTorch na Debian 12, trebate imati instaliran Python 3 PIP i Python virtual okruženje (venv).
Najprije ažurirajte predmemoriju repozitorija APT paketa sljedećom naredbom:
$ sudo prikladno ažuriranje 
Da biste instalirali Python 3 PIP i Python 3 virtualno okruženje (venv), pokrenite sljedeću naredbu:
$ sudo prikladan instalirati python3-pip python3-venv python3-dev Za potvrdu instalacije pritisnite “Y” i zatim pritisnite
Instaliraju se Python 3 PIP i Python 3 venv. Potrebno je neko vrijeme da se završi.
U ovom trenutku trebali bi biti instalirani Python 3 PIP i Python 3 venv.
Stvaranje Python 3 virtualnog okruženja za PyTorch
Standardna praksa za instaliranje Python biblioteka na Debian 12 je njihovo instaliranje u Python virtualnom okruženju tako da ne ometaju Python pakete/biblioteke sustava.
Da biste stvorili novo Python 3 virtualno okruženje za PyTorch u direktoriju “/opt/pytorch”, pokrenite sljedeću naredbu:
$ sudo python3 -m venv / opt / bakljaNadogradnja Python 3 PIP-a na najnoviju verziju na Python 3 PyTorch virtualnom okruženju
Da biste nadogradili Python 3 PIP na najnoviju verziju u Python 3 “/opt/pytorch” virtualnom okruženju, pokrenite sljedeću naredbu:
$ sudo / opt / baklja / kanta za smeće / pip3 instalirati --nadogradnja pip 
Instaliranje PyTorcha s NVIDIA GPU/CUDA podrškom za ubrzanje na Debian 12
Da bi PyTorch NVIDIA GPU/CUDA ubrzanje radilo, morate instalirati ispravnu verziju PyTorcha koja podržava verziju NVIDIA CUDA upravljačkog programa koju ste instalirali na svom Debian 12 sustavu. U vrijeme pisanja ovog teksta, PyTorch podržava NVIDIA CUDA drajver verzije 11.8 i 12.1. Za ažurirane informacije o verzijama upravljačkog programa NVIDIA CUDA koje PyTorch podržava, provjerite službenu web stranicu PyTorcha .
Da biste provjerili verziju upravljačkog programa NVIDIA CUDA koju ste instalirali na svoj Debian 12 sustav, pokrenite sljedeću naredbu. Kao što vidite, na našem Debian 12 sustavu imamo instaliranu NVIDIA CUDA verziju 11.8.
$ nvcc --verzija 
Da biste instalirali PyTorch s podrškom za NVIDIA CUDA 11.8 u virtualnom okruženju PyTorch Python 3, pokrenite sljedeću naredbu:
$ sudo / opt / baklja / kanta za smeće / pip3 instalirati baklja torchvision torchaudio --index-url https: // preuzimanje.pytorch.org / whl / sa 118Da biste instalirali PyTorch s podrškom za NVIDIA CUDA 12.1 u virtualnom okruženju PyTorch Python 3, pokrenite sljedeću naredbu:
$ sudo / opt / baklja / kanta za smeće / pip3 instalirati baklja torchvision torchaudioPyTorch se instalira na virtualno okruženje PyTorch Python 3. Potrebno je neko vrijeme da se završi.
U ovom trenutku, PyTorch bi trebao biti instaliran na PyTorch Python 3 virtualnom okruženju
Aktiviranje virtualnog okruženja PyTorch Python 3
Da biste aktivirali virtualno okruženje PyTorch Python “/opt/pytorch”, pokrenite sljedeću naredbu:
$ . / opt / baklja / kanta za smeće / aktiviratiTrebalo bi biti aktivirano virtualno okruženje PyTorch Python 3.
Pristup PyTorchu i provjera je li dostupno NVIDIA GPU/CUDA ubrzanje
Da biste otvorili Python 3 interaktivnu ljusku, pokrenite sljedeću naredbu:
$ python3Treba otvoriti interaktivnu ljusku Python 3.
Prvo uvezite PyTorch sa sljedećim retkom koda:
$ uvozna baklja 
Da biste provjerili verziju PyTorcha koju ste instalirali, pokrenite sljedeći redak koda. Kao što vidite, koristimo PyTorch 2.1.0 s podrškom za ubrzanje NVIDIA CUDA 11.8 (cu118).
$ baklja.__verzija__ 
Kako biste provjerili može li PyTorch koristiti vaš NVIDIA GPU za NVIDIA CUDA ubrzanje, možete pokrenuti i sljedeći redak koda. Ako je dostupna podrška za NVIDIA CUDA, bit će ispisano 'True'.
$ baklja.cuda.is_dostupan ( )Ako imate više GPU-ova instaliranih na vašem računalu, možete provjeriti broj GPU-ova koje PyTorch može koristiti pomoću sljedećeg retka koda. Kao što vidite, imamo NVIDIA GPU (RTX 4070) instaliran na našem Debian 12 sustavu.
$ baklja.cuda.broj_uređaja ( )Da biste izašli iz Python interaktivne ljuske, pokrenite sljedeći redak koda:
$ prestati ( )Zaključak
U ovom članku smo vam pokazali kako instalirati Python 3 PIP i Python 3 virtualno okruženje (venv) na Debian 12. Također smo vam pokazali kako stvoriti Python 3 virtualno okruženje za PyTorch na Debian 12 i kako instalirati PyTorch s NVIDIA CUDA Podrška za ubrzanje 11.8 i 12.1 također na Debianu 12. Na kraju smo vam pokazali kako aktivirati virtualno okruženje PyTorch Python i pristupiti PyTorchu na Debianu 12.