PyTorch може да ускори AI/ML приложенията с помощта на NVIDIA GPU чрез NVIDIA CUDA библиотеката, точно както TensorFlow.
В тази статия ще ви покажем как да инсталирате PyTorch с поддръжка на NVIDIA GPU/CUDA ускорение на Debian 12 „Bookworm“.
Тема на съдържанието:
- Инсталиране на NVIDIA GPU драйвери на Debian 12
- Инсталиране на NVIDIA CUDA на Debian 12
- Инсталиране на Python 3 PIP и Python 3 Virtual Environment (venv) на Debian 12
- Създаване на Python 3 виртуална среда за PyTorch
- Надграждане на Python 3 PIP до най-новата версия във виртуалната среда на Python 3 PyTorch
- Инсталиране на PyTorch с NVIDIA GPU/CUDA Acceleration Support на Debian 12
- Активиране на виртуалната среда PyTorch Python 3
- Достъп до PyTorch и проверка дали NVIDIA GPU/CUDA ускорението е налично
- Заключение
Инсталиране на NVIDIA GPU драйвери на Debian 12
За да работи PyTorch NVIDIA GPU/CUDA ускорението, трябва инсталирайте драйверите на NVIDIA GPU на Debian 12 . Ако имате нужда от помощ при инсталирането на драйверите на NVIDIA GPU на вашата система Debian 12, прочетете тази статия .
Инсталиране на NVIDIA CUDA на Debian 12
За да работи PyTorch NVIDIA GPU/CUDA ускорението на Debian 12, трябва инсталирайте NVIDIA CUDA на Debian 12 . Ако имате нужда от помощ при инсталирането на NVIDIA CUDA на вашата система Debian 12, прочетете тази статия .
Инсталиране на Python 3 PIP и Python 3 Virtual Environment (venv) на Debian 12
За да инсталирате PyTorch на Debian 12, трябва да имате инсталирани Python 3 PIP и виртуална среда Python (venv).
Първо актуализирайте кеша на хранилището на APT пакети със следната команда:
$ sudo подходяща актуализация 
За да инсталирате Python 3 PIP и Python 3 виртуална среда (venv), изпълнете следната команда:
$ sudo ап Инсталирай python3-pip python3-venv python3-dev За да потвърдите инсталацията, натиснете “Y” и след това натиснете
Инсталират се Python 3 PIP и Python 3 venv. Отнема известно време, за да завършите.
В този момент трябва да бъдат инсталирани Python 3 PIP и Python 3 venv.
Създаване на Python 3 виртуална среда за PyTorch
Стандартната практика за инсталиране на библиотеките на Python в Debian 12 е инсталирането им във виртуална среда на Python, така че да не пречат на пакетите/библиотеките на Python на системата.
За да създадете нова виртуална среда на Python 3 за PyTorch в директорията „/opt/pytorch“, изпълнете следната команда:
$ sudo python3 -м venv / избирам / pytorchНадграждане на Python 3 PIP до най-новата версия във виртуалната среда на Python 3 PyTorch
За да надстроите Python 3 PIP до най-новата версия във виртуалната среда на Python 3 „/opt/pytorch“, изпълнете следната команда:
$ sudo / избирам / pytorch / кошче / pip3 Инсталирай --подобряване на пип 
Инсталиране на PyTorch с NVIDIA GPU/CUDA Acceleration Support на Debian 12
За да работи PyTorch NVIDIA GPU/CUDA ускорението, трябва да инсталирате правилната версия на PyTorch, която поддържа версията на драйвера NVIDIA CUDA, която сте инсталирали на вашата система Debian 12. Към момента на писане на тази статия PyTorch поддържа драйвера NVIDIA CUDA версии 11.8 и 12.1. За актуализирана информация относно версиите на драйверите на NVIDIA CUDA, които PyTorch поддържа, проверете официалния уебсайт на PyTorch .
За да проверите версията на драйвера на NVIDIA CUDA, която сте инсталирали на вашата система Debian 12, изпълнете следната команда. Както можете да видите, имаме инсталирана NVIDIA CUDA версия 11.8 на нашата система Debian 12.
$ nvcc --версия 
За да инсталирате PyTorch с поддръжка на NVIDIA CUDA 11.8 във виртуалната среда PyTorch Python 3, изпълнете следната команда:
$ sudo / избирам / pytorch / кошче / pip3 Инсталирай факел torchvision torchaudio --index-url https: // download.pytorch.org / whl / със 118За да инсталирате PyTorch с поддръжка на NVIDIA CUDA 12.1 във виртуалната среда PyTorch Python 3, изпълнете следната команда:
$ sudo / избирам / pytorch / кошче / pip3 Инсталирай факел torchvision torchaudioPyTorch се инсталира във виртуалната среда PyTorch Python 3. Отнема известно време, за да завършите.
В този момент PyTorch трябва да бъде инсталиран във виртуалната среда PyTorch Python 3
Активиране на виртуална среда PyTorch Python 3
За да активирате виртуалната среда на PyTorch Python “/opt/pytorch”, изпълнете следната команда:
$ . / избирам / pytorch / кошче / активиратеВиртуалната среда PyTorch Python 3 трябва да бъде активирана.
Достъп до PyTorch и проверка дали NVIDIA GPU/CUDA ускорението е налично
За да отворите интерактивната обвивка на Python 3, изпълнете следната команда:
$ python3Трябва да се отвори интерактивната обвивка на Python 3.
Първо импортирайте PyTorch със следния ред код:
$ факел за внос 
За да проверите версията на PyTorch, която сте инсталирали, изпълнете следния ред код. Както можете да видите, работим с PyTorch 2.1.0 с поддръжка на ускорение NVIDIA CUDA 11.8 (cu118).
$ факла.__версия__ 
За да проверите дали PyTorch може да използва вашия NVIDIA GPU за NVIDIA CUDA ускорение, можете да изпълните и следния ред код. Ако е налична поддръжка на NVIDIA CUDA, ще бъде отпечатано „True“.
$ torch.cuda.is_available ( )Ако имате няколко графични процесора, инсталирани на вашия компютър, можете да проверите броя на графичните процесори, които PyTorch може да използва със следния ред код. Както можете да видите, ние имаме NVIDIA GPU (RTX 4070), инсталиран на нашата система Debian 12.
$ torch.cuda.device_count ( )За да излезете от интерактивната обвивка на Python, изпълнете следния ред код:
$ напусни ( )Заключение
В тази статия ви показахме как да инсталирате Python 3 PIP и Python 3 виртуална среда (venv) на Debian 12. Също така ви показахме как да създадете Python 3 виртуална среда за PyTorch на Debian 12 и как да инсталирате PyTorch с NVIDIA CUDA Поддръжка на ускорение 11.8 и 12.1 и на Debian 12. Накрая ви показахме как да активирате виртуалната среда на PyTorch Python и да получите достъп до PyTorch на Debian 12.