Ativar o PyTorch com DirectML no Windows

O PyTorch com DirectML permite treino e inferência em GPUs compatíveis com DirectX 12. O PyTorch com DirectML está em pré-visualização pública e funciona no Windows nativo a partir do Windows 10 versão 1709.

Verifica a tua versão do Windows

Para verificar a versão do Windows e o número de compilação, selecione a tecla + do logótipodo Windows R, enter winver, e selecione OK. Atualize para o Windows 10 versão 1709 ou posterior se a sua versão for mais antiga.

Verifique se há atualizações de driver de GPU

Instale o driver mais recente disponível para a sua GPU através do Windows Update ou do site do fabricante do seu hardware.

Configurar o Python

Instala um ambiente Python. Se usares o Miniconda, descarrega e executa o instalador do Windows para a tua arquitetura.

Depois cria e ativa um ambiente chamado pytorch-directml:

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

Instalar o PyTorch com DirectML

Instale o pacote torch-directml:

pip install torch-directml

Verifique a instalação

Inicie Python e execute o seguinte código para adicionar dois tensores no dispositivo 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())

Produção esperada:

3

Exemplos e feedback

Veja os exemplos do DirectML PyTorch para exemplos. Para reportar problemas de pacotes ou solicitar funcionalidades, utilize o rastreador de problemas do DirectML.