Aktifkan PyTorch dengan DirectML di Windows

PyTorch dengan DirectML memungkinkan pelatihan dan inferensi pada GPU berkemampuan DirectX 12. PyTorch dengan DirectML tersedia dalam pratinjau publik dan berfungsi secara natif di Windows mulai dari Windows 10 versi 1709.

Periksa versi Windows Anda

Untuk memeriksa versi Windows dan nomor build Anda, pilih kunci + logo WindowsR, masukkan winver, dan pilih OK. Perbarui ke Windows 10 versi 1709 atau yang lebih baru jika build Anda lebih lama.

Periksa pembaruan driver GPU

Instal driver terbaru yang tersedia untuk GPU Anda melalui Windows Update atau situs web produsen perangkat keras Anda.

Menyiapkan Python

Instal lingkungan Python. Jika Anda menggunakan Miniconda, unduh dan jalankan alat penginstal Windows untuk arsitektur Anda.

Kemudian buat dan aktifkan lingkungan bernama pytorch-directml:

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

Menginstal PyTorch dengan DirectML

Instal paket torch-directml:

pip install torch-directml

Memverifikasi penginstalan

Mulai Python dan jalankan kode berikut untuk menambahkan dua tensor pada perangkat 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())

Output yang diharapkan:

3

Sampel dan umpan balik

Lihat sampel DirectML PyTorch misalnya. Untuk melaporkan masalah paket atau fitur permintaan, gunakan pelacak masalah DirectML.