PyTorch dokáže zrýchliť aplikácie AI/ML pomocou GPU NVIDIA prostredníctvom knižnice NVIDIA CUDA natívne rovnako ako TensorFlow.
V tomto článku vám ukážeme, ako nainštalovať PyTorch s podporou akcelerácie NVIDIA GPU/CUDA na Debian 12 “Bookworm”.
Téma obsahu:
- Inštalácia ovládačov GPU NVIDIA na Debian 12
- Inštalácia NVIDIA CUDA na Debian 12
- Inštalácia Pythonu 3 PIP a virtuálneho prostredia Python 3 (venv) na Debian 12
- Vytvorenie virtuálneho prostredia Python 3 pre PyTorch
- Inovácia Python 3 PIP na najnovšiu verziu vo virtuálnom prostredí Python 3 PyTorch
- Inštalácia PyTorch s podporou akcelerácie NVIDIA GPU/CUDA na Debian 12
- Aktivácia virtuálneho prostredia PyTorch Python 3
- Prístup k PyTorch a kontrola, či je k dispozícii akcelerácia NVIDIA GPU/CUDA
- Záver
Inštalácia ovládačov GPU NVIDIA na Debian 12
Aby akcelerácia PyTorch NVIDIA GPU/CUDA fungovala, musíte nainštalujte ovládače GPU NVIDIA na Debian 12 . Ak potrebujete pomoc pri inštalácii ovládačov GPU NVIDIA na vašom systéme Debian 12, prečítajte si tento článok .
Inštalácia NVIDIA CUDA na Debian 12
Aby akcelerácia PyTorch NVIDIA GPU/CUDA fungovala na Debiane 12, musíte nainštalujte NVIDIA CUDA na Debian 12 . Ak potrebujete pomoc pri inštalácii NVIDIA CUDA na vašom systéme Debian 12, prečítajte si tento článok .
Inštalácia Pythonu 3 PIP a virtuálneho prostredia Python 3 (venv) na Debian 12
Ak chcete nainštalovať PyTorch na Debian 12, musíte mať nainštalované Python 3 PIP a virtuálne prostredie Python (venv).
Najprv aktualizujte vyrovnávaciu pamäť balíka APT pomocou nasledujúceho príkazu:
$ sudo apt update 
Ak chcete nainštalovať Python 3 PIP a virtuálne prostredie Python 3 (venv), spustite nasledujúci príkaz:
$ sudo apt Inštalácia python3-pip python3-venv python3-dev Na potvrdenie inštalácie stlačte „Y“ a potom stlačte
Inštaluje sa Python 3 PIP a Python 3 venv. Dokončenie chvíľu trvá.
V tomto bode by mali byť nainštalované Python 3 PIP a Python 3 venv.
Vytvorenie virtuálneho prostredia Python 3 pre PyTorch
Štandardnou praxou inštalácie knižníc Python na Debian 12 je ich inštalácia vo virtuálnom prostredí Pythonu, aby nezasahovali do systémových balíkov/knižníc Pythonu.
Ak chcete vytvoriť nové virtuálne prostredie Python 3 pre PyTorch v adresári „/opt/pytorch“, spustite nasledujúci príkaz:
$ sudo python3 -m venv / opt / pytorchInovácia Python 3 PIP na najnovšiu verziu vo virtuálnom prostredí Python 3 PyTorch
Ak chcete inovovať Python 3 PIP na najnovšiu verziu vo virtuálnom prostredí Python 3 „/opt/pytorch“, spustite nasledujúci príkaz:
$ sudo / opt / pytorch / kôš / pip3 Inštalácia -- upgrade pip 
Inštalácia PyTorch s podporou akcelerácie NVIDIA GPU/CUDA na Debian 12
Aby akcelerácia PyTorch NVIDIA GPU/CUDA fungovala, musíte nainštalovať správnu verziu PyTorch, ktorá podporuje verziu ovládača NVIDIA CUDA, ktorú ste nainštalovali do systému Debian 12. V čase písania tohto článku PyTorch podporuje ovládače NVIDIA CUDA verzie 11.8 a 12.1. Aktualizované informácie o verziách ovládačov NVIDIA CUDA, ktoré PyTorch podporuje, pozrite si oficiálnu webovú stránku PyTorch .
Ak chcete skontrolovať verziu ovládača NVIDIA CUDA, ktorú ste nainštalovali do systému Debian 12, spustite nasledujúci príkaz. Ako vidíte, na našom systéme Debian 12 máme nainštalovanú verziu NVIDIA CUDA 11.8.
$ nvcc --verzia 
Ak chcete nainštalovať PyTorch s podporou NVIDIA CUDA 11.8 vo virtuálnom prostredí PyTorch Python 3, spustite nasledujúci príkaz:
$ sudo / opt / pytorch / kôš / pip3 Inštalácia baterka torchvision torchaudio --index-url https: // download.pytorch.org / whl / so 118Ak chcete nainštalovať PyTorch s podporou NVIDIA CUDA 12.1 vo virtuálnom prostredí PyTorch Python 3, spustite nasledujúci príkaz:
$ sudo / opt / pytorch / kôš / pip3 Inštalácia baterka torchvision torchaudioPyTorch sa inštaluje do virtuálneho prostredia PyTorch Python 3. Dokončenie chvíľu trvá.
V tomto bode by mal byť PyTorch nainštalovaný vo virtuálnom prostredí PyTorch Python 3
Aktivácia virtuálneho prostredia PyTorch Python 3
Ak chcete aktivovať virtuálne prostredie PyTorch Python „/opt/pytorch“, spustite nasledujúci príkaz:
$ . / opt / pytorch / kôš / AktivovaťVirtuálne prostredie PyTorch Python 3 by malo byť aktivované.
Prístup k PyTorch a kontrola, či je k dispozícii akcelerácia NVIDIA GPU/CUDA
Ak chcete otvoriť interaktívny shell Python 3, spustite nasledujúci príkaz:
$ python3Interaktívny shell Pythonu 3 by sa mal otvoriť.
Najprv importujte PyTorch s nasledujúcim riadkom kódu:
$ dovozová baterka 
Ak chcete skontrolovať verziu PyTorch, ktorú ste nainštalovali, spustite nasledujúci riadok kódu. Ako môžete vidieť, používame PyTorch 2.1.0 s podporou akcelerácie NVIDIA CUDA 11.8 (cu118).
$ pochodeň.__verzia__ 
Ak chcete skontrolovať, či PyTorch dokáže použiť váš GPU NVIDIA na zrýchlenie NVIDIA CUDA, môžete spustiť aj nasledujúci riadok kódu. Ak je k dispozícii podpora NVIDIA CUDA, vytlačí sa „True“.
$ torch.cuda.is_available ( )Ak máte vo svojom počítači nainštalovaných viacero GPU, počet GPU, ktoré môže PyTorch použiť, môžete skontrolovať pomocou nasledujúceho riadku kódu. Ako vidíte, na našom systéme Debian 12 máme nainštalovaný grafický procesor NVIDIA (RTX 4070).
$ torch.cuda.device_count ( )Ak chcete ukončiť interaktívny shell Pythonu, spustite nasledujúci riadok kódu:
$ skončiť ( )Záver
V tomto článku sme vám ukázali, ako nainštalovať virtuálne prostredie Python 3 PIP a Python 3 (venv) na Debian 12. Tiež sme vám ukázali, ako vytvoriť virtuálne prostredie Python 3 pre PyTorch na Debian 12 a ako nainštalovať PyTorch pomocou NVIDIA CUDA Podpora zrýchlenia 11.8 a 12.1 aj na Debiane 12. Nakoniec sme vám ukázali, ako aktivovať virtuálne prostredie PyTorch Python a získať prístup k PyTorch na Debian 12.