python - 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

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

升级版：

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