Aktivera PyTorch med DirectML på Windows

PyTorch med DirectML möjliggör träning och slutsatsdragning på DirectX 12-kompatibla GPU:er. PyTorch med DirectML är i offentlig förhandsversion och fungerar på interna Windows från och med Windows 10 version 1709.

Kontrollera din version av Windows

Om du vill kontrollera din Windows version och versionsnummer väljer du Windows tangenten + R, anger winveroch väljer OK. Uppdatera till Windows 10 version 1709 eller senare om din version är äldre.

Sök efter GPU-drivrutinsuppdateringar

Installera den senaste drivrutinen som är tillgänglig för din GPU via Windows Update eller maskinvarutillverkarens webbplats.

Konfigurera Python

Installera en Python miljö. Om du använder Miniconda laddar du ned och kör installationsprogrammet Windows för din arkitektur.

Skapa och aktivera sedan en miljö med namnet pytorch-directml:

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

Installera PyTorch med DirectML

Installera torch-directml-paketet:

pip install torch-directml

Kontrollera installationen

Starta Python och kör följande kod för att lägga till två tensorer på DirectML-enheten:

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())

Förväntat resultat:

3

Exempel och feedback

Exempel finns i DirectML PyTorch-exempel . Om du vill rapportera paketproblem eller begära funktioner använder du DirectML-problemspåraren.