如何在Google Colab上安装PyTorch v1.0.0+？

Question

PyTorch v1.0.0稳定版在2018年12月8日发布, 在此之前提前7个月宣布。

我想获得针对我的IPython内核运行的硬件进行优化的版本。

我该如何在Google Colab上获取此版本？

- Tom Hale

Answer 1

尝试以下代码片段（无论是否使用GPU运行时均有效）

!pip install -q torch==1.0.0 torchvision

检查版本号

import torch
print(torch.__version__)

这里是版本1.0.0。

更新

!pip install torch

现在正常运作，因为最稳定的版本是1.0.0

- ashunigion

---

这会考虑系统上运行的CUDA版本进行优化吗？ - Tom Hale

我认为它是可以的，我尝试了torch.backends.cudnn.version()，输出结果是7401，而且torch.backends.cudnn.enabled == True的输出结果是true。我使用的是Colab GPU运行时。 - ashunigion

太好了！即使我的解决方案基于Google的colab代码片段并且在生成的下载URL中有cu90，我也得到了7410。 - Tom Hale

Answer 2

从版本1.0.0开始，PyTorch将下载URL格式更改为：

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

to

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变量。

- Tom Hale

Answer 3

现在你可以只需要：

import torch

无需额外安装。

- Linh Le Vu

Answer 4

对于1.1.0版本，这个可以实现。最初的回答。

!pip install -q torch==1.1.0 torchvision

- Oladimeji Mudele

Answer 5

对我很有效，你可以试试

!pip install torch
!pip install torchvision
!pip install mxnet-cu101

- Maheep

---

实际上，当我使用前两个命令安装PyTorch时，在导入时出现了错误，但是当我安装mxnet-cu101后，它就可以工作了。 - Maheep

Answer 6

这是安装PyTorch的代码。您可以将其更改为所需版本。

!pip3 install http://download.pytorch.org/whl/cu92/torch-1.0.0-cp36-cp36m-linux_x86_64.whl

- Payam Jome Yazdian