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?
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
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.
Pour la version 1.1.0, cela fonctionne
!pip install -q torch==1.1.0 torchvision
Vous pouvez maintenant simplement
import torch
Pas besoin d'installation supplémentaire.