pipenv install pytorch cpu + 具体版本

Question

我需要安装特定版本的 pytorch cpu 模式。

使用 pip 我会这样做：

pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html --trusted-host download.pytorch.org

如何使用 Pipenv 实现同样的效果？

我尝试使用以下 Pipfile：

[[source]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cpu"
verify_ssl = false

[packages]
torchvision = {index = "pytorch", version = "==0.4.0"}
torch = {index = "pytorch", version = "==1.2.0"}

但没用

Answer 1

你可以这样做：

$ PIP_FIND_LINKS="https://download.pytorch.org/whl/torch_stable.html" pipenv install torch==1.2.0+cpu torchvision==0.4.0+cpu

但是，您必须确保为任何连续的pipenv sync、pipenv lock 等添加PIP_FIND_LINKS。

更新：

您也可以将PIP_FIND_LINKS="https://download.pytorch.org/whl/torch_stable.html" 添加到.env 文件中，但它只会在pipenv run 和pipenv shell 上加载。