为 CUDA 9.0 安装 PyTorch 时遇到问题

Trouble installing PyTorch for CUDA 9.0

我正在尝试为 CUDA 9.0 (Ubuntu 18.04 LTS) 安装 PyTorch。我按照 official documentation 的说明进行操作,上面写着

从以下 html 页面下载所需版本的 whl 文件: https://download.pytorch.org/whl/cu90/torch_stable.html

$ pip install [downloaded file]

但是上面link坏了,打不开

使用

中的 link

https://pytorch.org/

这是获得您想要的东西的最简单、最好的方法。

如果 cuda 9.0 无法运行,请尝试非 cuda 或 cuda 9.2

根据official documentation可以用命令安装pytorch

# for Python 3.x
pip3 install torch torchvision

或者如果你喜欢 Python 2

# for Python 2.x
pip install torch torchvision

原来 https://download.pytorch.org/whl/cu90/torch_stable.html<a href> 标签的列表,我们可以通过浏览器进入源代码来查看。然后我们可以下载合适的版本并通过 pip install [downloaded file].

安装