Neste tutorial, abordaremos o processo de instalação do PyTorch em sistemas Windows e Linux, utilizando o Anaconda para a gestão dos pacotes necessários.
A instalação será realizada por meio da ferramenta de linha de comando `conda`, integrada ao Anaconda.
PyTorch é uma biblioteca de aprendizado de máquina, escrita em Python e fundamentada no framework Torch. Desenvolvida pelo Facebook, ela se compara ao TensorFlow do Google. Sua aplicação é vasta em áreas como visão computacional e processamento de linguagem natural, sendo utilizada por empresas como a Tesla no desenvolvimento de software para condução autônoma.
O PyTorch é um projeto de código aberto, gratuito, licenciado sob a licença BSD modificada e sob a Linux Foundation.
Requisitos Preliminares
Para acompanhar este guia, é essencial que você tenha o Anaconda instalado no seu computador.
Caso ainda não o tenha feito, consulte este guia sobre como instalar o Anaconda, que o orientará por todo o procedimento. Após concluir a instalação do Anaconda, você poderá prosseguir com a instalação do PyTorch.
Instalando o PyTorch no Linux
Como prática recomendada, comece por atualizar os pacotes de software na sua distribuição Linux. No meu caso, estou utilizando o Ubuntu e o `apt` para gerenciar os pacotes, então utilizarei o seguinte comando para atualizar:
$ sudo apt update && apt upgrade
Após a conclusão da atualização dos pacotes, acesse a página oficial de instalação do PyTorch. Deslize a página até encontrar o assistente de instalação, que se parece com o exemplo a seguir:
Com este assistente, você poderá clicar nas diferentes opções, informando as preferências do seu sistema. Em contrapartida, você receberá um comando que poderá usar no seu terminal para instalar o PyTorch.
Após inserir as informações do meu sistema, a configuração ficou da seguinte maneira:
Optei pela versão estável para Linux e utilizarei o Conda para gerenciar os pacotes. Também escolhi usar o PyTorch com Python, em vez de C++/Java, e executarei o PyTorch em uma CPU, e não em uma GPU.
Na parte inferior da tabela está o comando que posso usar para instalar o PyTorch. No entanto, antes de executá-lo, irei criar um ambiente virtual Anaconda nomeado `pytorch`.
Os ambientes virtuais possibilitam a criação de projetos e o isolamento de suas dependências de outros projetos, evitando conflitos. Uma das vantagens do Anaconda é a facilidade com que ele permite criar e gerenciar ambientes virtuais.
Para criar um ambiente virtual com a versão 3.7 do Python, utilizarei o seguinte comando:
conda create -n pytorch python=3.7
Após a criação do ambiente, vou ativá-lo com o seguinte comando:
conda activate pytorch
Com o ambiente ativo, executarei o comando gerado anteriormente no site do PyTorch para instalar a biblioteca.
conda install pytorch torchvision torchaudio cpuonly -c pytorch
Siga as instruções para instalar o PyTorch. Após a instalação, reinicie a sessão do terminal para que as alterações tenham efeito.
Para verificar se o PyTorch foi instalado corretamente, vamos tentar importá-lo no shell interativo do Python. Certifique-se de estar no ambiente virtual `pytorch` usando o comando:
conda activate pytorch
Estando no ambiente virtual, abra o shell interativo do Python com o seguinte comando:
python
Com a sessão do shell iniciada, digite a seguinte linha de código e pressione ENTER:
import torch
Se o Python for executado sem erros, a instalação foi bem-sucedida. Caso contrário, se você receber um erro de módulo não encontrado, algo ocorreu errado durante a instalação e você pode tentar reinstalar o PyTorch.
Instalando o PyTorch no Windows
Inicialmente, no seu sistema Windows, procure pelo programa “Anaconda Prompt” e execute-o. É nesse ambiente que executaremos os comandos.
Com o programa aberto, vamos criar um ambiente virtual para a instalação do PyTorch utilizando o seguinte comando:
conda create -n pytorch python=3.7
Após a criação do ambiente virtual, podemos ativá-lo com o seguinte comando:
conda activate pytorch
Com o ambiente virtual ativo, podemos prosseguir para a instalação do PyTorch. Para isso, vamos acessar a página de instalação no site oficial do PyTorch e deslizar até a seção onde se localiza o assistente de instalação:
Nesta seção, selecionamos as informações do nosso sistema e o assistente nos fornecerá um comando para instalar o PyTorch. Selecionarei a versão estável para Windows, gerenciada pelo Conda, utilizando a linguagem de programação Python e executada em uma CPU. Como resultado, minha tabela ficará da seguinte forma:
Em seguida, copie o comando fornecido, cole-o no prompt do Anaconda e pressione ENTER.
Após a conclusão da instalação, podemos verificar se ela ocorreu corretamente abrindo o shell interativo do Python e tentando importar o PyTorch.
Portanto, dentro do prompt do Anaconda, inicie uma sessão interativa do Python.
python
Com a sessão iniciada, importe o PyTorch utilizando a seguinte linha de código:
import torch
Se esta ação for concluída sem erros, a instalação foi bem-sucedida.
Considerações Finais
Neste guia, instalamos o PyTorch no Windows e Linux utilizando o Conda. Também é possível instalá-lo via PIP, como um pacote comum do PIP. Em ambos os casos, optei pela instalação em CPU. No entanto, você ainda pode usar o CUDA, um kit de ferramentas da Nvidia que acelera o treinamento por meio da paralelização de operações em GPUs.