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 AI Toolkit for VS Code.
Controllare la versione di Windows
Il pacchetto torch-directml in Windows nativo funziona a partire da Windows 10 versione 1709 (Build 16299 o successiva). È possibile controllare il numero di versione della build eseguendo winver tramite il comando Esegui (tasto logo Windows + R).
Verificare la disponibilità di aggiornamenti dei driver GPU
Assicurarsi di avere installato il driver GPU più recente. Selezionare Controlla la disponibilità di aggiornamenti nella sezione Windows Update dell'app Impostazioni.
Configurare Torch-DirectML
È consigliabile configurare un ambiente Python virtuale in Windows. Sono disponibili molti strumenti che è possibile usare per configurare un ambiente Python virtuale. Per queste istruzioni si userà miniconda di Anaconda. Il resto di questa configurazione presuppone che si usi un ambiente Miniconda.
Il pacchetto torch-directml supporta fino a PyTorch 2.2
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 di Python e importare Torch con le righe seguenti:
La versione corrente di torch-directml viene mappata al back-end "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)
Aggiungere i tensori insieme e stampare i risultati.
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.
Imparare i concetti chiave alla base della creazione di modelli di Machine Learning con PyTorch. Eseguire il training di un modello di rete neurale che riconosce e classifica le immagini.
Illustrare i concetti fondamentali dell'intelligenza artificiale correlati allo sviluppo di software e servizi di Microsoft Azure per creare soluzioni di intelligenza artificiale.