Povolení PyTorchu s DirectML na Windows

PyTorch s DirectML nabízí vývojářům snadný způsob, jak vyzkoušet nejnovější a nejlepší modely AI na svém Windows počítači. PyTorch s DirectML si můžete stáhnout instalací balíčku PyPi torch-directml. Po nastavení můžete začít s našimi ukázkami nebo použít Sadu nástrojů Foundry pro VS Code.

Zkontrolujte svou verzi Windows

Balíček torch-directml na nativních Windows funguje od Windows 10 verze 1709 (build 16299 nebo vyšší). Číslo verze sestavení můžete zkontrolovat spuštěním příkazu winver prostřednictvím příkazu Run (Windows klávesa s logem + R).

Kontrola aktualizací ovladačů GPU

Ujistěte se, že máte nainstalovaný nejnovější ovladač GPU. Vyberte možnost Vyhledat aktualizace v části služba Windows Update nastavení systému Windows Settings.

Nastavte Torch-DirectML

Doporučujeme nastavit virtuální Python prostředí uvnitř Windows. Existuje mnoho nástrojů, které můžete použít k nastavení virtuálního prostředí Pythonu – pro tyto pokyny použijeme Miniconda od Anaconda. Zbývající část tohoto nastavení předpokládá, že používáte prostředí Miniconda.

Nastavení prostředí Python

Stáhněte a nainstalujte instalační program Miniconda Windows do systému. Na webu Anaconda najdete další pokyny k nastavení . Po instalaci Miniconda vytvořte prostředí pomocí Python s názvem pytdml a aktivujte ho pomocí následujících příkazů.

conda create --name pytdml -y
conda activate pytdml

Nainstalujte PyTorch a Torch-DirectML

Poznámka:

Balíček torch-directml podporuje až PyTorch 2.3.1.

Vše, co je potřeba k instalaci, je instalace nejnovější verze torch-directml spuštěním následujícího příkazu:

pip install torch-directml

Ověření a vytvoření zařízení

Po instalaci balíčku torch-directml můžete ověřit, že se správně spustí přidáním dvou tensorů. Nejprve spusťte interaktivní relaci v Pythonu a importujte Torch s následujícími řádky:

import torch
import torch_directml
dml = torch_directml.device()

Aktuální verze torch-directml je přiřazena k backendu Torch "PrivateUse1". Rozhraní API torch_directml.device() je pohodlný obal pro odesílání tensorů do zařízení DirectML.

S vytvořeným zařízením DirectML teď můžete definovat dva jednoduché tensory; jeden tenzor obsahující 1 a druhý obsahující 2. Umístěte tenzory na zařízení "dml".

tensor1 = torch.tensor([1]).to(dml) # Note that dml is a variable, not a string!
tensor2 = torch.tensor([2]).to(dml)

Přidejte tensory dohromady a vytiskněte výsledky.

dml_algebra = tensor1 + tensor2
dml_algebra.item()

Měl by se zobrazit výstup čísla 3, jak je znázorněno v následujícím příkladu.

>>> import torch
>>> tensor1 = torch.tensor([1]).to(dml)
>>> tensor2 = torch.tensor([2]).to(dml)
>>> dml_algebra = tensor1 + tensor2
>>> dml_algebra.item()
3

PyTorch s ukázkami DirectML a zpětnou vazbou

Podívejte se na naše ukázky a podívejte se na další použití PyTorch s DirectML. Pokud narazíte na problémy nebo máte zpětnou vazbu k balíčku PyTorch s balíčkem DirectML, spojte se sem s naším týmem.