Activer PyTorch avec DirectML sur Windows

PyTorch avec DirectML permet l’apprentissage et l’inférence sur des GPU compatibles DirectX 12. PyTorch avec DirectML est en préversion publique et fonctionne sur des Windows natifs à partir de Windows 10 version 1709.

Vérifiez votre version de Windows

Pour vérifier votre version de Windows et votre numéro de build, appuyez sur la touche du logo Windows + R, entrez winver, puis sélectionnez OK. Effectuez une mise à jour vers Windows 10 version 1709 ou ultérieure si votre build est antérieure.

Rechercher les mises à jour du pilote GPU

Installez le pilote le plus récent disponible pour votre GPU via Windows Update ou le site web de votre fabricant de matériel.

Configurer Python

Installez un environnement Python. Si vous utilisez Miniconda, téléchargez et exécutez le programme d’installation Windows pour votre architecture.

Ensuite, créez et activez un environnement nommé pytorch-directml:

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

Installer PyTorch avec DirectML

Installez le package torch-directml :

pip install torch-directml

Vérifier l’installation

Démarrez Python et exécutez le code suivant pour ajouter deux tenseurs sur l’appareil 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())

Sortie attendue :

3

Exemples et commentaires

Pour obtenir des exemples, consultez les exemples PyTorch DirectML . Pour signaler des problèmes de package ou des fonctionnalités de demande, utilisez le suivi des problèmes DirectML.