Versões pré-compiladas de Pytorch

· 1 min

Resumo #

Como usar o repo torch_stable do PyTorch para instalar versões antigas ou específicas direto pelo pip.

Encontrando versões no pip #

Acabo de descobrir um repositório que salvou minha pele, então resolvi vir aqui compartilhar com vocês.

Se você abrir o site do pytorch, ele te apresenta algumas opções de instalação:

Opções de instalação exibidas no site do PyTorch

A questão é que, se você quiser instalar uma versão diferente da apresentada (hoje é a 1.7.1), vai começar uma caça ao tesouro bem chatinha…

Existe uma página - que sei lá o porquê me levou um tempo pra achar - aqui:

https://download.pytorch.org/whl/torch_stable.html

Mas não é necessário abrir esse link! Você pode pedir ao pip pra buscar pra você:

pip[3] install pytorch==1.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

E voilá, seu pacote vai ser instalado automagicamente ;)

Dicas extras #

Algumas dicas extras:

  1. Algumas versões de cuda são consideradas “default”. Por exemplo, se você procurar por 1.6.0+cu102 não vai ter lá, porque pip install cuda==1.6.0 vai buscar automaticamente a versão com cuda 10.2.

  2. Se for masoquista e quiser instalar uma versão CPU-only: pip install 1.6.0+cpu

  3. Para salvar esse repositório no seu requirements.txt, basta adicionar como primeira linha:

-f https://download.pytorch.org/whl/torch_stable.html

...
torch==1.6.0+cu110
...

Happy torching!