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

Question

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

我想为运行IPython内核的硬件优化版本。

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

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

Answer 2

在 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在发布新版本时根据需要更改变量。