Abilitare PyTorch con DirectML in Windows

PyTorch con DirectML abilita il training e l'inferenza nelle GPU con supporto per DirectX 12. PyTorch con DirectML è disponibile in anteprima pubblica e funziona in Windows nativa a partire da Windows 10 versione 1709.

Controllare la versione di Windows

Per controllare la versione Windows e il numero di build, selezionare Windows tasto + logoR, immettere winvere selezionare OK. Eseguire l'aggiornamento a Windows 10 versione 1709 o successiva se la build è precedente.

Verificare la disponibilità di aggiornamenti dei driver GPU

Installare il driver più recente disponibile per la GPU tramite Windows Update o il sito Web del produttore dell'hardware.

Configurare Python

Installare un ambiente Python. Se si utilizza Miniconda, scaricare ed eseguire il programma di installazione di Windows per la propria architettura.

Creare e attivare quindi un ambiente denominato pytorch-directml:

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

Installare PyTorch con DirectML

Installare il pacchetto torch-directml:

pip install torch-directml

Verificare l'installazione

Avviare Python ed eseguire il codice seguente per aggiungere due tensori nel dispositivo DirectML:

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

Output previsto:

3

Esempi e feedback

Per alcuni esempi, vedere gli esempi DirectML per PyTorch. Per segnalare i problemi del pacchetto o le funzionalità di richiesta, usare lo strumento di rilevamento dei problemi directML.