PyTorch voi nopeuttaa AI/ML-sovelluksia käyttämällä NVIDIA GPU:ta NVIDIA CUDA -kirjaston kautta natiivisti, kuten TensorFlow.
Tässä artikkelissa näytämme, kuinka PyTorch asennetaan NVIDIA GPU/CUDA -kiihdytyksen tuella Debian 12 'Bookworm' -versioon.
Sisällön aihe:
- NVIDIA GPU -ajurien asentaminen Debian 12:een
- NVIDIA CUDA:n asentaminen Debian 12:een
- Python 3 PIP:n ja Python 3 Virtual Environmentin (venv) asentaminen Debian 12:een
- Python 3 -virtuaaliympäristön luominen PyTorchille
- Python 3 PIP:n päivittäminen uusimpaan versioon Python 3 PyTorch -virtuaaliympäristössä
- PyTorchin asentaminen NVIDIA GPU/CUDA Acceleration Support -tuella Debian 12:ssa
- PyTorch Python 3 -virtuaaliympäristön aktivointi
- PyTorchiin pääsy ja sen tarkistaminen, onko NVIDIA GPU/CUDA Acceleration saatavilla
- Johtopäätös
NVIDIA GPU -ajurien asentaminen Debian 12:een
PyTorch NVIDIA GPU/CUDA -kiihdytys toimii, sinun on asenna NVIDIA GPU -ohjaimet Debian 12:een . Jos tarvitset apua NVIDIA GPU -ohjainten asentamisessa Debian 12 -järjestelmääsi, lue tämä artikkeli .
NVIDIA CUDA:n asentaminen Debian 12:een
Jotta PyTorch NVIDIA GPU/CUDA -kiihdytys toimisi Debian 12:ssa, sinun on asenna NVIDIA CUDA Debian 12:een . Jos tarvitset apua NVIDIA CUDA:n asentamisessa Debian 12 -järjestelmääsi, lue tämä artikkeli .
Python 3 PIP:n ja Python 3 Virtual Environmentin (venv) asentaminen Debian 12:een
PyTorchin asentaminen Debian 12:een edellyttää, että Python 3 PIP ja Python-virtuaaliympäristö (venv) on asennettu.
Päivitä ensin APT-pakettivaraston välimuisti seuraavalla komennolla:
$ sudo osuva päivitys 
Asenna Python 3 PIP ja Python 3 virtuaaliympäristö (venv) suorittamalla seuraava komento:
$ sudo apt Asentaa python3-pip python3-venv python3-dev Vahvista asennus painamalla 'Y' ja sitten paina
Python 3 PIP ja Python 3 venv asennetaan. Valmistuminen kestää jonkin aikaa.
Tässä vaiheessa Python 3 PIP ja Python 3 venv tulisi asentaa.
Python 3 -virtuaaliympäristön luominen PyTorchille
Vakiokäytäntö Python-kirjastojen asentamisessa Debian 12:een on asentaa ne Python-virtuaaliympäristöön, jotta ne eivät häiritse järjestelmän Python-paketteja/kirjastoja.
Luo uusi Python 3 -virtuaaliympäristö PyTorchille '/opt/pytorch' -hakemistoon suorittamalla seuraava komento:
$ sudo python3 -m venv / valita / pytorchPython 3 PIP:n päivittäminen uusimpaan versioon Python 3 PyTorch -virtuaaliympäristössä
Päivitä Python 3 PIP viimeisimpään versioon Python 3 '/opt/pytorch' -virtuaaliympäristössä suorittamalla seuraava komento:
$ sudo / valita / pytorch / roskakori / pip3 Asentaa --päivitys pip 
PyTorchin asentaminen NVIDIA GPU/CUDA Acceleration Support -tuella Debian 12:ssa
PyTorchin NVIDIA GPU/CUDA -kiihdytys toimisi, kun sinun on asennettava oikea PyTorchin versio, joka tukee Debian 12 -järjestelmääsi asentamaasi NVIDIA CUDA -ohjainversiota. Tätä kirjoitettaessa PyTorch tukee NVIDIA CUDA -ohjainversioita 11.8 ja 12.1. Saat päivitetyt tiedot PyTorchin tukemista NVIDIA CUDA -ohjainversioista: Tarkista PyTorchin virallinen verkkosivusto .
Tarkista Debian 12 -järjestelmääsi asentamasi NVIDIA CUDA -ohjainversio suorittamalla seuraava komento. Kuten näet, meillä on NVIDIA CUDA -versio 11.8 asennettuna Debian 12 -järjestelmäämme.
$ nvcc --versio 
Asenna PyTorch NVIDIA CUDA 11.8 -tuella PyTorch Python 3 -virtuaaliympäristöön suorittamalla seuraava komento:
$ sudo / valita / pytorch / roskakori / pip3 Asentaa taskulamppu torchvision torchaudio --index-url https: // download.pytorch.org / whl / 118:n kanssaAsenna PyTorch NVIDIA CUDA 12.1 -tuella PyTorch Python 3 -virtuaaliympäristöön suorittamalla seuraava komento:
$ sudo / valita / pytorch / roskakori / pip3 Asentaa taskulamppu torchvision torchaudioPyTorch asennetaan PyTorch Python 3 -virtuaaliympäristöön. Valmistuminen kestää jonkin aikaa.
Tässä vaiheessa PyTorch tulee asentaa PyTorch Python 3 -virtuaaliympäristöön
PyTorch Python 3 -virtuaaliympäristön aktivointi
Aktivoidaksesi PyTorch Python “/opt/pytorch” -virtuaaliympäristön, suorita seuraava komento:
$ . / valita / pytorch / roskakori / aktivoidaPyTorch Python 3 -virtuaaliympäristö tulee aktivoida.
PyTorchiin pääsy ja sen tarkistaminen, onko NVIDIA GPU/CUDA Acceleration saatavilla
Avaa Python 3:n interaktiivinen kuori suorittamalla seuraava komento:
$ python3Python 3:n interaktiivinen kuori tulee avata.
Tuo ensin PyTorch seuraavalla koodirivillä:
$ tuonti taskulamppu 
Voit tarkistaa asentamasi PyTorchin version suorittamalla seuraavan koodirivin. Kuten näet, käytämme PyTorch 2.1.0:aa NVIDIA CUDA 11.8 -kiihdytyksen tuella (cu118).
$ taskulamppu.__versio__ 
Voit tarkistaa, pystyykö PyTorch käyttämään NVIDIA GPU:ta NVIDIA CUDA -kiihdytykseen, suorittamalla myös seuraavan koodirivin. Jos NVIDIA CUDA -tuki on saatavilla, 'True' tulostetaan.
$ torch.cuda.is_available ( )Jos tietokoneellesi on asennettu useita GPU:ita, voit tarkistaa PyTorchin käyttämien GPU:iden määrän seuraavalla koodirivillä. Kuten näet, meillä on NVIDIA GPU (RTX 4070) asennettuna Debian 12 -järjestelmäämme.
$ torch.cuda.device_count ( )Poistuaksesi Pythonin interaktiivisesta kuoresta suorita seuraava koodirivi:
$ lopettaa ( )Johtopäätös
Tässä artikkelissa osoitimme, kuinka Python 3 PIP ja Python 3 -virtuaaliympäristö (venv) asennetaan Debian 12:een. Näimme myös, kuinka Python 3 -virtuaaliympäristö luodaan PyTorchille Debian 12:ssa ja kuinka PyTorch asennetaan NVIDIA CUDA:lla. 11.8- ja 12.1-kiihdytyksen tuki myös Debian 12:ssa. Lopuksi näytimme sinulle, kuinka PyTorch Python -virtuaaliympäristö aktivoidaan ja PyTorchiin pääsee Debian 12:ssa.