Ativar o PyTorch com DirectML no Windows

O PyTorch com DirectML fornece uma maneira fácil de usar para os desenvolvedores experimentarem os melhores e mais recentes modelos de IA em sua máquina Windows. Você pode baixar o PyTorch com DirectML instalando o pacote torch-directml PyPi. Uma vez configurado, você pode começar com nossos exemplos ou usar o AI Toolkit for VS Code.

Verifique a sua versão do Windows

O pacote torch-directml no Windows nativo funciona a partir do Windows 10, versão 1709 (Build 16299 ou superior). Você pode verificar o número da versão da compilação executando winver através do comando Executar (tecla do logotipo do Windows + R).

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

Certifique-se de ter o driver de GPU mais recente instalado. Selecione Verificar se há atualizações na seção Windows Update das Configurações do Windows.

Configurar Torch-DirectML

Recomendamos configurar um ambiente Python virtual dentro do Windows. Há muitas ferramentas que você pode usar para configurar um ambiente Python virtual — para estas instruções, usaremos o Miniconda do Anaconda. O restante dessa configuração pressupõe que você use um ambiente Miniconda.

Configurar um ambiente Python

Baixe e instale o instalador Miniconda do Windows em seu sistema. Há orientações adicionais para configuração no site da Anaconda. Uma vez instalado o Miniconda, crie um ambiente usando Python chamado pytdml e ative-o através dos seguintes comandos.

conda create --name pytdml -y
conda activate pytdml

Instale o PyTorch e Torch-DirectML

Tudo o que é necessário para obter a instalação é instalar a versão mais recente do torch-directml executando o seguinte comando:

pip install torch-directml

Verificação e criação de dispositivos

Depois de instalar o pacote torch-directml , você pode verificar se ele é executado corretamente adicionando dois tensores. Primeiro, inicie uma sessão interativa do Python e importe o Torch com as seguintes linhas:

import torch
import torch_directml
dml = torch_directml.device()

A versão atual do torch-directml é mapeada para o backend da tocha "PrivateUse1". A API torch_directml.device() é um wrapper conveniente para enviar seus tensores para o dispositivo DirectML.

Com o dispositivo DirectML criado, agora você pode definir dois tensores simples; um tensor contendo um 1 e outro contendo um 2. Coloque os tensores no dispositivo denominado "dml".

tensor1 = torch.tensor([1]).to(dml) # Note that dml is a variable, not a string!
tensor2 = torch.tensor([2]).to(dml)

Adicione os tensores e imprima os resultados.

dml_algebra = tensor1 + tensor2
dml_algebra.item()

Deve ver o número 3 sendo exibido, como no exemplo abaixo.

>>> import torch
>>> tensor1 = torch.tensor([1]).to(dml)
>>> tensor2 = torch.tensor([2]).to(dml)
>>> dml_algebra = tensor1 + tensor2
>>> dml_algebra.item()
3

PyTorch com amostras e comentários do DirectML

Confira nossos exemplos para ver mais usos do PyTorch com DirectML. Se você tiver problemas, ou tiver feedback sobre o pacote PyTorch com DirectML, entre em contato com nossa equipe aqui.