PyTorch var paātrināt AI/ML lietojumprogrammas, izmantojot NVIDIA GPU, izmantojot NVIDIA CUDA bibliotēku, tāpat kā TensorFlow.
Šajā rakstā mēs parādīsim, kā instalēt PyTorch ar NVIDIA GPU/CUDA paātrinājuma atbalstu Debian 12 “Bookworm”.
Satura tēma:
- NVIDIA GPU draiveru instalēšana operētājsistēmā Debian 12
- NVIDIA CUDA instalēšana operētājsistēmā Debian 12
- Python 3 PIP un Python 3 virtuālās vides (venv) instalēšana operētājsistēmā Debian 12
- Python 3 virtuālās vides izveide PyTorch
- Python 3 PIP jaunināšana uz jaunāko versiju Python 3 PyTorch virtuālajā vidē
- PyTorch instalēšana ar NVIDIA GPU/CUDA paātrinājuma atbalstu operētājsistēmā Debian 12
- PyTorch Python 3 virtuālās vides aktivizēšana
- Piekļuve PyTorch un pārbaude, vai ir pieejams NVIDIA GPU/CUDA paātrinājums
- Secinājums
NVIDIA GPU draiveru instalēšana operētājsistēmā Debian 12
Lai PyTorch NVIDIA GPU/CUDA paātrinājums darbotos, jums tas ir jādara instalējiet NVIDIA GPU draiverus operētājsistēmā Debian 12 . Ja jums nepieciešama palīdzība NVIDIA GPU draiveru instalēšanā Debian 12 sistēmā, izlasi šo rakstu .
NVIDIA CUDA instalēšana operētājsistēmā Debian 12
Lai PyTorch NVIDIA GPU/CUDA paātrinājums darbotos ar Debian 12, jums ir instalējiet NVIDIA CUDA uz Debian 12 . Ja jums nepieciešama palīdzība, instalējot NVIDIA CUDA savā Debian 12 sistēmā, izlasi šo rakstu .
Python 3 PIP un Python 3 virtuālās vides (venv) instalēšana operētājsistēmā Debian 12
Lai instalētu PyTorch operētājsistēmā Debian 12, ir jāinstalē Python 3 PIP un Python virtuālā vide (venv).
Vispirms atjauniniet APT pakotnes repozitorija kešatmiņu ar šādu komandu:
$ sudo piemērots atjauninājums 
Lai instalētu Python 3 PIP un Python 3 virtuālo vidi (venv), palaidiet šo komandu:
$ sudo apt uzstādīt python3-pip python3-venv python3-dev Lai apstiprinātu instalēšanu, nospiediet “Y” un pēc tam nospiediet
Tiek instalēts Python 3 PIP un Python 3 venv. Tas aizņem kādu laiku, lai pabeigtu.
Šajā brīdī ir jāinstalē Python 3 PIP un Python 3 venv.
Python 3 virtuālās vides izveide PyTorch
Standarta prakse Python bibliotēku instalēšanai Debian 12 ir to instalēšana Python virtuālajā vidē, lai tās netraucētu sistēmas Python pakotnēm/bibliotēkām.
Lai izveidotu jaunu Python 3 virtuālo vidi PyTorch direktorijā “/opt/pytorch”, palaidiet šādu komandu:
$ sudo python3 -m venv / izvēlēties / pytorchPython 3 PIP jaunināšana uz jaunāko versiju Python 3 PyTorch virtuālajā vidē
Lai jauninātu Python 3 PIP uz jaunāko versiju Python 3 “/opt/pytorch” virtuālajā vidē, palaidiet šo komandu:
$ sudo / izvēlēties / pytorch / atkritumu tvertne / pip3 uzstādīt -- jauninājums pip 
PyTorch instalēšana ar NVIDIA GPU/CUDA paātrinājuma atbalstu operētājsistēmā Debian 12
Lai PyTorch NVIDIA GPU/CUDA paātrinājums darbotos, jums ir jāinstalē pareizā PyTorch versija, kas atbalsta NVIDIA CUDA draivera versiju, kuru instalējāt savā Debian 12 sistēmā. Šīs rakstīšanas laikā PyTorch atbalsta NVIDIA CUDA draivera versijas 11.8 un 12.1. Lai iegūtu atjauninātu informāciju par PyTorch atbalstītajām NVIDIA CUDA draiveru versijām, pārbaudiet PyTorch oficiālo vietni .
Lai pārbaudītu NVIDIA CUDA draivera versiju, ko instalējāt savā Debian 12 sistēmā, palaidiet šo komandu. Kā redzat, mūsu Debian 12 sistēmā ir instalēta NVIDIA CUDA versija 11.8.
$ nvcc -- versija 
Lai PyTorch Python 3 virtuālajā vidē instalētu PyTorch ar NVIDIA CUDA 11.8 atbalstu, palaidiet šo komandu:
$ sudo / izvēlēties / pytorch / atkritumu tvertne / pip3 uzstādīt lāpa torchvision torchaudio --index-url https: // download.pytorch.org / whl / ar 118Lai PyTorch Python 3 virtuālajā vidē instalētu PyTorch ar NVIDIA CUDA 12.1 atbalstu, palaidiet šo komandu:
$ sudo / izvēlēties / pytorch / atkritumu tvertne / pip3 uzstādīt lāpa torchvision torchaudioPyTorch tiek instalēta PyTorch Python 3 virtuālajā vidē. Tas aizņem kādu laiku, lai pabeigtu.
Šajā brīdī PyTorch ir jāinstalē PyTorch Python 3 virtuālajā vidē
PyTorch Python 3 virtuālās vides aktivizēšana
Lai aktivizētu PyTorch Python “/opt/pytorch” virtuālo vidi, palaidiet šādu komandu:
$ . / izvēlēties / pytorch / atkritumu tvertne / aktivizētIr jāaktivizē PyTorch Python 3 virtuālā vide.
Piekļuve PyTorch un pārbaude, vai ir pieejams NVIDIA GPU/CUDA paātrinājums
Lai atvērtu Python 3 interaktīvo čaulu, palaidiet šādu komandu:
$ python3Ir jāatver Python 3 interaktīvais apvalks.
Vispirms importējiet PyTorch ar šādu koda rindu:
$ importa lāpa 
Lai pārbaudītu instalēto PyTorch versiju, palaidiet šo koda rindiņu. Kā redzat, mēs darbinām PyTorch 2.1.0 ar NVIDIA CUDA 11.8 paātrinājuma atbalstu (cu118).
$ lāpa.__versija__ 
Lai pārbaudītu, vai PyTorch spēj izmantot jūsu NVIDIA GPU NVIDIA CUDA paātrināšanai, varat palaist arī šādu koda rindiņu. Ja ir pieejams NVIDIA CUDA atbalsts, tiks izdrukāts “True”.
$ torch.cuda.is_available ( )Ja datorā ir instalēti vairāki GPU, varat pārbaudīt PyTorch izmantoto GPU skaitu, izmantojot tālāk norādīto koda rindiņu. Kā redzat, mūsu Debian 12 sistēmā ir instalēts NVIDIA GPU (RTX 4070).
$ torch.cuda.device_count ( )Lai izietu no Python interaktīvās čaulas, palaidiet šo koda rindiņu:
$ atmest ( )Secinājums
Šajā rakstā mēs parādījām, kā instalēt Python 3 PIP un Python 3 virtuālo vidi (venv) Debian 12. Mēs arī parādījām, kā izveidot Python 3 virtuālo vidi PyTorch operētājsistēmā Debian 12 un kā instalēt PyTorch ar NVIDIA CUDA. 11.8 un 12.1 paātrinājuma atbalsts arī operētājsistēmā Debian 12. Visbeidzot, mēs parādījām, kā aktivizēt PyTorch Python virtuālo vidi un piekļūt PyTorch operētājsistēmā Debian 12.