Блог

Мне нужно установить определенную версию pytorch cpu mode.

С пипом я бы сделал это так:

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?

Я попытался создать следующий Pip-файл:

 [[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"}
 

но это не сработало

Вы можете сделать:

 $ 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 , и т.д.

UPD:

Вы также можете добавить PIP_FIND_LINKS="https://download.pytorch.org/whl/torch_stable.html" в .env файл, но он загружается только на pipenv run и pipenv shell .