PyTorch met DirectML inschakelen op Windows

PyTorch met DirectML maakt training en deductie mogelijk voor Gpu's die geschikt zijn voor DirectX 12. PyTorch met DirectML is beschikbaar als openbare previewversie en werkt rechtstreeks onder Windows vanaf Windows 10, versie 1709.

Controleer uw versie van Windows

Als u uw Windows-versie en buildnummer wilt controleren, selecteert u Windows logotoets + R, voert u in winveren selecteert u OK. Werk bij naar Windows 10 versie 1709 of hoger als uw build ouder is.

Controleren op updates van GPU-stuurprogramma's

Installeer het meest recente stuurprogramma dat beschikbaar is voor uw GPU via Windows Update of de website van de hardwarefabrikant.

Python instellen

Installeer een Python-omgeving. Als u Miniconda gebruikt, downloadt en voert u het Windows-installatieprogramma voor uw architectuur uit.

Maak en activeer vervolgens een omgeving met de naam pytorch-directml:

conda create --name pytorch-directml python=3.10
conda activate pytorch-directml

PyTorch installeren met DirectML

Installeer het torch-directml-pakket:

pip install torch-directml

De installatie controleren

Start Python en voer de volgende code uit om twee tensors toe te voegen op het DirectML-apparaat:

import torch
import torch_directml

dml = torch_directml.device()
tensor1 = torch.tensor([1]).to(dml)
tensor2 = torch.tensor([2]).to(dml)
dml_algebra = tensor1 + tensor2
print(dml_algebra.item())

Verwachte uitvoer:

3

Voorbeelden en feedback

Zie de Voorbeelden van DirectML PyTorch voor voorbeelden. Als u pakketproblemen of aanvraagfuncties wilt melden, gebruikt u de DirectML-probleemtracker.