Aktivieren von PyTorch mit DirectML auf Windows

PyTorch mit DirectML ermöglicht Schulungen und Rückschlüsse auf DirectX 12-fähige GPUs. PyTorch mit DirectML befindet sich in der öffentlichen Vorschau und arbeitet ab Windows 10 Version 1709 an nativen Windows.

Überprüfen Sie die Windows-Version

Um Ihre Windows Version und Buildnummer zu überprüfen, wählen Sie Windows Logo-Taste + R aus, geben Sie winver"OK" ein, und wählen Sie "OK" aus. Aktualisieren Sie auf Windows 10 Version 1709 oder höher, wenn Ihr Build älter ist.

Überprüfen auf GPU-Treiberupdates

Installieren Sie den neuesten Treiber, der für Ihre GPU über Windows Update oder die Website Ihres Hardwareherstellers verfügbar ist.

Einrichten von Python

Installieren Sie eine Python Umgebung. Wenn Sie Miniconda verwenden, laden Sie das Windows Installer für Ihre Architektur herunter, und führen Sie es aus.

Erstellen und aktivieren Sie dann eine Umgebung mit dem Namen pytorch-directml:

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

Installieren von PyTorch mit DirectML

Installieren Sie das Paket torch-directml:

pip install torch-directml

Überprüfen der Installation

Starten Sie Python, und führen Sie den folgenden Code aus, um zwei Tensoren auf dem DirectML-Gerät hinzuzufügen:

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

Erwartete Ausgabe:

3

Beispiele und Feedback

Beispiele finden Sie in den DirectML PyTorch-Beispielen . Verwenden Sie die DirectML-Problemverfolgung, um Paketprobleme oder Anforderungsfeatures zu melden.