Instale e configure o PyTorch em seu computador.

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

Obter o PyTorch

Primeiro, configure um ambiente do Python.

É recomendável configurar um ambiente virtual do Python dentro do Windows, usando o Anaconda como um gerenciador de pacotes. O restante desta configuração pressupõe que você esteja usando um ambiente do 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 dá suporte apenas ao Python 3.x; não há suporte para o Python 2.x.

Anaconda installers

Depois que a instalação for concluída, 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

Anaconda command line prompt

Agora, você pode instalar o pacote PyTorch de binários por meio do 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
  • Linguagem – Python
  • Plataforma de Computação – CPU ou escolha sua versão do Cuda. Neste tutorial, você treinará e usará um modelo de inferência na CPU, mas também poderá usar uma GPU Nvidia.

Anaconda setup

  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

Anaconda installation

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

Confirm Anaconda installation

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

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

Depois, insira o seguinte código:

import torch 

x = torch.rand(2, 3) 

print(x)

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

Verify Pytorch functionality

Nota

Quer aprender mais? Visite o site oficial do PyTorch

Próximas etapas

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