从 requirements.txt 安装 PyTorch

Question

Torch 文档说使用

pip install torch==1.4.0+cpu torchvision==0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

安装最新版本的 PyTorch。这在我手动执行时有效，但是当我将其添加到 req.txt 并执行时pip install -r req.txt，它失败并显示 ERROR: No matching distribution.

编辑：从 req.txt 添加整行并在此处添加错误。

torch==1.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

torchvision==0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.htmltorch==1.4.0+cpu

ERROR: Could not find a version that satisfies the requirement torch==1.4.0+cpu (from -r requirements.txt (line 1)) (from versions: 0.1.2, 0.1.2.post1, 0.1.2.post2, 0.3.1, 0.4.0, 0.4.1, 1.0.0, 1.0.1, 1.0.1.post2, 1.1.0, 1.2.0, 1.3.0, 1.3.1, 1.4.0)
ERROR: No matching distribution found for torch==1.4.0+cpu (from -r requirements.txt (line 1))

Answer 1

--find-links在requirements.txt火炬前添加

--find-links https://download.pytorch.org/whl/torch_stable.html

torch==1.2.0+cpu

来源：https : //github.com/pytorch/pytorch/issues/29745#issuecomment-553588171

Answer 2

为了获得我需要的 cuda 版本（而不是存储库提供的任何版本），我从pytorch转换了特定于 cuda 的安装命令：

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

在requirements.txt的顶部：

--extra-index-url https://download.pytorch.org/whl/cu116
torch
torchvision
torchaudio

然后我像平常一样pip install -r requirements.txt，当我在 python 中import torch运行时torch.version.cuda，我得到了我想要的“11.6”。

Answer 3

-f https://download.pytorch.org/whl/torch_stable.html 
torch==1.4.0+cpu 
-f https://download.pytorch.org/whl/torch_stable.html
torchvision==0.5.0+cpu

对我来说效果很好:)

Answer 4

你可以这样做：

$pip install -r req.txt --find-links https://download.pytorch.org/whl/torch_stable.html

只需将您的 PyTorch 要求输入如下req.txt：

torch==1.4.0+cpu

torchvision==0.5.0+cpu