web-dev-qa-db-fra.com

Comment installer PyTorch v1.0.0 + sur Google Colab?

PyTorch v1.0.0 stable était publié le 8 décembre 2018 après avoir été annoncé 7 mois plus tôt .

Je veux obtenir une version optimisée pour le matériel sur lequel mon noyau IPython fonctionne.

Comment obtenir cette version sur Google Colab?

7
Tom Hale

essayez l'extrait de code suivant (il fonctionne également pour le runtime avec ou sans gpu)

!pip install -q torch==1.0.0 torchvision

vérifier la version

import torch
print(torch.__version__)

vous avez ici la version 1.0.0

MISE À JOUR

!pip install torch

Fonctionne très bien maintenant, car la version la plus stable est 1.0.0

9
ashunigion

Avec la version 1.0.0, PyTorch a changé le format d'URL de téléchargement de:

https://download.pytorch.org/whl/cu92/torch-1.0.0-cp36-cp36m-linux_x86_64.whl

à

https://download.pytorch.org/whl/cu90/torch-1.0.0-cp36-cp36m-linux_x86_64.whl

(Le changement se trouve dans la partie de la version CUDA, où cu92 devient cu90.)

Pour générer par programme cette URL, j'ai utilisé le code suivant:

from os.path import exists
from wheel.pep425tags import get_abbr_impl, get_impl_ver, get_abi_tag

platform = '{}{}-{}'.format(get_abbr_impl(), get_impl_ver(), get_abi_tag())
cuda_output = !ldconfig -p|grep cudart.so|sed -e 's/.*\.\([0-9]*\)\.\([0-9]*\)$/cu\10/'    
accelerator = cuda_output[0] if exists('/dev/nvidia0') else 'cpu'

torch_url=f"http://download.pytorch.org/whl/{accelerator}/torch-{version}-{platform}-linux_x86_64.whl"
version='1.0.0'

!pip install -U {torch_url} torchvision

Vous pouvez ensuite modifier la variable version comme vous le souhaitez à mesure que de nouvelles versions sont publiées.

3
Tom Hale

Pour la version 1.1.0, cela fonctionne

!pip install -q torch==1.1.0 torchvision
2
Dimeji Mudele

Vous pouvez maintenant simplement

import torch

Pas besoin d'installation supplémentaire.

1
Lê Vũ Linh