Я пытаюсь установить указанную c версию pytorch, совместимую с указанной c версией драйвера cuda с pipenv. На веб-сайте pytorch показано, как это сделать с помощью pip:
pip3 install torch==1.3.1+cu92 torchvision==0.4.2+cu92 -f https://download.pytorch.org/whl/torch_stable.html
Я попытался преобразовать это в запись в моем Pipfile, например:
[[source]]
name = "pytorch"
url = "https://download.pytorch.org/whl/torch_stable.html"
verify_ssl = false
pytorch = {version="==1.3.1+cu92", index="pytorch"}
torchvision = {version="==0.4.2+cu92", index="pytorch"}
Однако это не работает. Зависимость с этой версией не может быть решена. Я не уверен, что URL, указанный в параметре -f в команде pip3, является даже допустимым источником для pipenv.
Я мог бы установить обе библиотеки, просто передав команду pip следующим образом:
pipenv run pip install torch==1.3.1+cu92 torchvision==0.4.2+cu92 -f https://download.pytorch.org/whl/torch_stable.html
, но я не очень доволен этим решением, поскольку зависимости отсутствуют в Pipfile и я должен вручную документировать использование этой команды.