Windows üzerinde DirectML ile PyTorch'i etkinleştirme

DirectML ile PyTorch, DirectX 12 özellikli GPU'larda eğitim ve çıkarım sağlar. DirectML ile PyTorch genel önizleme aşamasındadır ve Windows 10 sürüm 1709'dan başlayarak yerel Windows üzerinde çalışır.

Windows sürümünüzü denetleyin

Windows sürümünüzü ve derleme numaranızı denetlemek için Windows logo tuşu + R'yi seçin, girin winverve Tamam'ı seçin. Derlemeniz daha eskiyse Windows 10 sürüm 1709 veya sonraki bir sürüme güncelleştirin.

GPU sürücü güncelleştirmelerini denetleme

Windows Update veya donanım üreticinizin web sitesi aracılığıyla GPU'nuz için kullanılabilen en son sürücüyü yükleyin.

Python'ı ayarlayın

bir Python ortamı yükleyin. Miniconda kullanıyorsanız mimariniz için Windows yükleyicisini indirip çalıştırın.

Ardından adlı pytorch-directmlbir ortam oluşturup etkinleştirin:

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

DirectML ile PyTorch yükleme

torch-directml Paketi yükleyin:

pip install torch-directml

Yüklemeyi doğrulama

Python başlatın ve DirectML cihazına iki tensor eklemek için aşağıdaki kodu çalıştırın:

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

Beklenen çıkış:

3

Örnekler ve geri bildirim

Örnekler için bkz. DirectML PyTorch örnekleri . Paket sorunlarını veya istek özelliklerini bildirmek için DirectML sorun izleyicisini kullanın.