PyTorch saab kiirendada AI/ML-rakendusi, kasutades NVIDIA GPU-d NVIDIA CUDA teegi kaudu, nagu TensorFlow.
Selles artiklis näitame teile, kuidas installida PyTorch koos NVIDIA GPU/CUDA kiirenduse toega Debian 12 'Bookworm' jaoks.
Sisu teema:
- NVIDIA GPU draiverite installimine Debian 12-sse
- NVIDIA CUDA installimine Debian 12-le
- Python 3 PIP ja Python 3 virtuaalse keskkonna (venv) installimine Debian 12-sse
- Python 3 virtuaalse keskkonna loomine PyTorchi jaoks
- Python 3 PIP uuendamine Python 3 PyTorchi virtuaalkeskkonnas uusimale versioonile
- PyTorchi installimine koos NVIDIA GPU/CUDA kiirenduse toega Debian 12-s
- PyTorch Python 3 virtuaalse keskkonna aktiveerimine
- Juurdepääs PyTorchile ja kontrollimine, kas NVIDIA GPU/CUDA kiirendus on saadaval
- Järeldus
NVIDIA GPU draiverite installimine Debian 12-sse
Et PyTorch NVIDIA GPU/CUDA kiirendus töötaks, peate seda tegema installige Debian 12-sse NVIDIA GPU draiverid . Kui vajate abi NVIDIA GPU draiverite installimisel oma Debian 12 süsteemi, lugege seda artiklit .
NVIDIA CUDA installimine Debian 12-le
Et PyTorch NVIDIA GPU/CUDA kiirendus töötaks Debian 12 puhul, peate installige NVIDIA CUDA Debian 12-le . Kui vajate abi NVIDIA CUDA installimisel oma Debian 12 süsteemi, lugege seda artiklit .
Python 3 PIP ja Python 3 virtuaalse keskkonna (venv) installimine Debian 12-sse
PyTorchi installimiseks Debian 12-le peab teil olema installitud Python 3 PIP ja Python virtuaalne keskkond (venv).
Esmalt värskendage APT pakettide hoidla vahemälu järgmise käsuga:
$ sudo sobiv värskendus 
Python 3 PIP ja Python 3 virtuaalse keskkonna (venv) installimiseks käivitage järgmine käsk:
$ sudo asjakohane installida python3-pip python3-venv python3-dev Installimise kinnitamiseks vajutage 'Y' ja seejärel vajutage
Python 3 PIP ja Python 3 venv installitakse. Selle valmimine võtab natuke aega.
Sel hetkel tuleks installida Python 3 PIP ja Python 3 venv.
Python 3 virtuaalse keskkonna loomine PyTorchi jaoks
Pythoni teekide installimise tavapärane praktika Debian 12-le on nende installimine Pythoni virtuaalsesse keskkonda, nii et need ei segaks süsteemi Pythoni pakette/teeke.
PyTorchi jaoks uue Python 3 virtuaalse keskkonna loomiseks kataloogis '/opt/pytorch' käivitage järgmine käsk:
$ sudo python3 -m venv / opt / pütorchPython 3 PIP uuendamine Python 3 PyTorchi virtuaalkeskkonnas uusimale versioonile
Python 3 PIP uuendamiseks Python 3 “/opt/pytorch” virtuaalkeskkonnas uusimale versioonile käivitage järgmine käsk:
$ sudo / opt / pütorch / prügikast / pip3 installida -- uuendada pip 
PyTorchi installimine koos NVIDIA GPU/CUDA kiirenduse toega Debian 12-s
PyTorchi NVIDIA GPU/CUDA kiirenduse toimimiseks peate installima PyTorchi õige versiooni, mis toetab oma Debian 12 süsteemi installitud NVIDIA CUDA draiveri versiooni. Selle kirjutamise ajal toetab PyTorch NVIDIA CUDA draiveri versioone 11.8 ja 12.1. Värskendatud teabe saamiseks PyTorchi toetatavate NVIDIA CUDA draiveriversioonide kohta vaadake PyTorchi ametlikku veebisaiti .
Debian 12 süsteemi installitud NVIDIA CUDA draiveri versiooni kontrollimiseks käivitage järgmine käsk. Nagu näete, on meie Debian 12 süsteemi installitud NVIDIA CUDA versioon 11.8.
$ nvcc -- versioon 
NVIDIA CUDA 11.8 toega PyTorchi installimiseks PyTorch Python 3 virtuaalsesse keskkonda käivitage järgmine käsk:
$ sudo / opt / pütorch / prügikast / pip3 installida tõrvik tõrvik tõrvikaudio --index-url https: // download.pytorch.org / whl / 118-gaNVIDIA CUDA 12.1 toega PyTorchi installimiseks PyTorch Python 3 virtuaalsesse keskkonda käivitage järgmine käsk:
$ sudo / opt / pütorch / prügikast / pip3 installida tõrvik tõrvik tõrvikaudioPyTorch installitakse PyTorch Python 3 virtuaalsesse keskkonda. Selle valmimine võtab natuke aega.
Sel hetkel tuleks PyTorch PyTorch Python 3 virtuaalsesse keskkonda installida
PyTorch Python 3 virtuaalse keskkonna aktiveerimine
PyTorch Pythoni virtuaalse keskkonna '/opt/pytorch' aktiveerimiseks käivitage järgmine käsk:
$ . / opt / pütorch / prügikast / aktiveeridaPyTorch Python 3 virtuaalne keskkond peaks olema aktiveeritud.
Juurdepääs PyTorchile ja kontrollimine, kas NVIDIA GPU/CUDA kiirendus on saadaval
Python 3 interaktiivse kesta avamiseks käivitage järgmine käsk:
$ python3Python 3 interaktiivne kest tuleks avada.
Esiteks importige PyTorch järgmise koodireaga:
$ import tõrvik 
Installitud PyTorchi versiooni kontrollimiseks käivitage järgmine koodirida. Nagu näete, käitame PyTorch 2.1.0 koos NVIDIA CUDA 11.8 kiirenduse toega (cu118).
$ taskulamp.__version__ 
Kontrollimaks, kas PyTorch suudab teie NVIDIA GPU-d NVIDIA CUDA kiirendamiseks kasutada, võite käivitada ka järgmise koodirea. Kui NVIDIA CUDA tugi on saadaval, prinditakse 'True'.
$ torch.cuda.is_available ( )Kui teie arvutisse on installitud mitu GPU-d, saate kontrollida PyTorchi kasutatavate GPU-de arvu järgmise koodireaga. Nagu näete, on meie Debian 12 süsteemi installitud NVIDIA GPU (RTX 4070).
$ torch.cuda.device_count ( )Pythoni interaktiivsest kestast väljumiseks käivitage järgmine koodirida:
$ lõpeta ( )Järeldus
Selles artiklis näitasime, kuidas installida Python 3 PIP ja Python 3 virtuaalne keskkond (venv) Debian 12-sse. Samuti näitasime, kuidas luua PyTorchi jaoks Python 3 virtuaalne keskkond Debian 12-s ja kuidas installida PyTorch koos NVIDIA CUDAga. 11.8 ja 12.1 kiirenduse tugi ka Debian 12-s. Lõpuks näitasime teile, kuidas aktiveerida PyTorch Pythoni virtuaalne keskkond ja pääseda juurde PyTorchile Debian 12-s.