Compartilhar via

Instalar e configurar o PyTorch

No estágio anterior deste tutorial, discutimos os conceitos básicos do PyTorch e os pré-requisitos de usá-lo para criar um modelo de machine learning. Aqui, vamos instalá-lo em seu computador.

Obter PyTorch

Primeiro, você precisará configurar um ambiente do Python.

Recomendamos configurar um ambiente virtual do Python dentro do Windows, usando o Anaconda como um gerenciador de pacotes. O restante dessa configuração pressupõe que você use um ambiente Anaconda.

  1. Baixe e instale o Anaconda aqui. Selecione Anaconda 64-bit installer for Windows Python 3.8.

Importante

Lembre-se de instalar o Python 3.x. Atualmente, o PyTorch no Windows só dá suporte ao Python 3.x; Não há suporte para Python 2.x.

Instaladores do Anaconda

Após a conclusão da instalação, verifique as versões do Anaconda e do Python.

  1. Abra o gerenciador do Anaconda em Iniciar – Anaconda3 – Prompt do Anaconda PowerShell e teste suas versões:

Você pode verificar sua versão do Python executando o seguinte comando: python –-version

Você pode verificar sua versão do Anaconda executando o seguinte comando: conda –-version

Prompt de linha de comando do Anaconda

Agora, você pode instalar o pacote PyTorch a partir de binários via Conda.

  1. Navegue até https://pytorch.org/.

Selecione os detalhes relevantes da instalação do PyTorch:

  • Build do PyTorch – estável.
  • Seu sistema operacional – Windows
  • Pacote – Conda
  • Idioma – Python
  • Plataforma de Computação – CPU ou escolha sua versão do Cuda. Neste tutorial, você treinará e inferirá o modelo na CPU, mas também poderá usar uma GPU Nvidia.
  1. Abra o gerenciador do Anaconda e execute o comando conforme especificado nas instruções de instalação.
conda install pytorch torchvision torchaudio cpuonly -c pytorch

Instalação do Anaconda

  1. Confirme e conclua a extração dos pacotes necessários.

Confirmar a instalação do Anaconda

Vamos verificar a instalação do PyTorch executando um código PyTorch de exemplo para construir um tensor inicializado aleatoriamente.

  1. Abra o prompt do Anaconda PowerShell e execute o comando a seguir.
python

Em seguida, insira o seguinte código:

import torch 

x = torch.rand(2, 3) 

print(x)

A saída deve ser um tensor 5x3 aleatório. Os números serão diferentes, mas devem ser semelhantes aos mostrados abaixo.

Verificar a funcionalidade do Pytorch

Observação

Quer aprender mais? Visite o site oficial do PyTorch

Próximas etapas

Agora que instalamos o PyTorch, estamos prontos para configurar os dados para nosso modelo.