我目前正在使用 Pipenv 来维护特定项目中使用的 Python 包。到目前为止,我尝试过的大多数下载都按预期工作;也就是我进入pipenv install [package],它把包安装到虚拟环境中,然后将包信息记录到Pipfile和Pipfile.lock中。
但是,我在安装 PyTorch 时遇到了一些问题。
我试过运行pipenv install torch,但每次锁定步骤失败。相反,我尝试使用以下命令直接从 PyTorch 网站强制下载
pipenv run pip install torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
Run Code Online (Sandbox Code Playgroud)
它实际上安装了!如果我运行pipenv graph它,它会同时显示 torch 和 torchvision 及其依赖项。但是仍然存在一个问题:torch 和 torchvision 都没有保存到 Pipfile 和 Pipfile.lock 中。
关于如何实现这一点的任何想法?