如何在Google Colab上安装pytorch v1.0.0+?

尝试以下代码段(它在运行时使用或不使用GPU时同样有效)

!pip install -q torch==1.0.0 torchvision

检查版本

import torch
print(torch.__version__)

这里有1.0.0版

更新

!pip install torch

现在工作正常,因为最稳定的版本是 1.0.0

对于版本1.0.0,pytorch更改了下载URL格式,从:

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

(更改在CUDA版本部分,其中 cu92 更改为 cu90 )

为了以编程方式生成该URL,我使用了以下代码:

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

然后你可以改变 version 随新版本的发布而变化。