Abilitare PyTorch con DirectML in WSL 2

PyTorch con DirectML offre agli sviluppatori un modo facile da usare per provare i modelli di intelligenza artificiale più recenti e più grandi nel computer Windows. È possibile scaricare PyTorch con DirectML installando il pacchetto PyPi torch-directml . Dopo aver configurato, è possibile iniziare con gli esempi o usare Foundry Toolkit per VS Code.

Controllare la versione di Windows

Il pacchetto torch-directml nel sottosistema Windows per Linux (WSL) 2 funziona a partire da Windows 11 (Build 22000 o versione successiva). È possibile controllare il numero di versione della build eseguendo winver tramite il comando Run (Windows tasto logo + R).

Verificare la disponibilità di aggiornamenti dei driver GPU

Assicurarsi di avere installato il driver GPU più recente. Selezionare Check for updates nella sezione Windows Update di Windows Settings.

Configurare Torch-DirectML

Installare WSL 2

Per installare il sottosistema Windows per Linux (WSL) 2, vedere le istruzioni in InstallA WSL.

Installare quindi il driver gui WSL seguendo le istruzioni nel file README.md nel repository microsoft/wslg GitHub.

Configurare un ambiente Python

È consigliabile configurare un ambiente di Python virtuale in WSL 2. Sono disponibili molti strumenti che è possibile usare per configurare un ambiente di Python virtuale. In questo argomento si userà Miniconda di Anaconda. Il resto di questa configurazione presuppone che si usi un ambiente Miniconda.

Installare Miniconda seguendo le indicazioni del programma di installazione di Linux nel sito di Anaconda o eseguendo i comandi seguenti in WSL 2.

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh 
bash Miniconda3-latest-Linux-x86_64.sh

Dopo aver installato Miniconda, creare un ambiente di Python denominato pytdml e attivarlo tramite i comandi seguenti:

conda create --name pytdml -y
conda activate pytdml

Installa PyTorch e Torch-DirectML

Tutto ciò che è necessario per ottenere l'installazione consiste nell'installare la versione più recente di torch-directml eseguendo il comando seguente:

pip install torch-directml

Verifica e creazione del dispositivo

Dopo aver installato il pacchetto torch-directml , è possibile verificare che venga eseguito correttamente aggiungendo due tensori. Avviare prima una sessione interattiva Python e importare Torch con le righe seguenti:

import torch
import torch_directml
dml = torch_directml.device()

La versione corrente di torch-directml è associata al backend "PrivateUse1". L'API torch_directml.device() è un wrapper pratico per inviare i tensori al dispositivo DirectML.

Dopo aver creato il dispositivo DirectML, è ora possibile definire due tensori semplici; un tensore contenente un valore 1 e un altro contenente un valore 2. Posizionare i tensori nel dispositivo "dml".

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

Aggiungi i tensori insieme e stampa i risultati.

dml_algebra = tensor1 + tensor2
dml_algebra.item()

Dovresti vedere visualizzato il numero 3, come nell'esempio seguente.

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

PyTorch con esempi e commenti di DirectML

Vedere gli esempi per visualizzare altri usi di PyTorch con DirectML. Se si verificano problemi o si hanno commenti e suggerimenti sul pacchetto PyTorch con DirectML, contattare il team qui.