Python 诗歌与火把

我最近发现诗歌有助于解决我们团队的所有不同依赖关系。 在一个项目中，我们也使用PyTorch，我想知道是否有人可以帮助我，并为我指出如何将其添加到诗歌中的方向。 我们正在使用无法访问CUDA GPU（用于简单的道路推断/测试）的机器和能够访问CUDA CPU的工作站。现在我想使用诗歌来确保每个开发人员在诗歌的帮助下使用相同的版本，但这可能吗

似乎没有明显的方法来决定安装哪个PyTorch版本。我曾考虑添加不同的安装说明作为额外的依赖项，但我找不到一个选项来获得等效的设置，如：

pip3安装火炬==1.3.1+cpu火炬视觉==0.4.2+cpu-fhttps://download.pytorch.org/whl/torch_stable.html

我可以设置不同在线控制盘的总路径，如：

https://download.pytorch.org/whl/torch_stable.html/cpu/torch-1.3.1%2Bcpu-cp36-cp36m-win_amd64.whl

但我宁愿不直接把它们放在git中。。。我在诗歌中看到的最接近的选项是手动下载它们，然后使用

file=X

命令

---

如果有任何帮助，我将不胜感激。：）

目前，Poetry没有

-f

选项（有an和an），因此不能使用

pip

指令。您可以直接安装

.whl

文件：

poetry add https://download.pytorch.org/whl/torch_stable.html/cpu/torch-1.3.1%2Bcpu-cp36-cp36m-win_amd64.whl

或者直接将依赖项添加到

.toml

文件中：

[tool.poetry.dependencies]
torch = { url = "https://download.pytorch.org/whl/torch_stable.html/cpu/torch-1.3.1%2Bcpu-cp36-cp36m-win_amd64.whl" }

github中的更新解决方案：

poetry add torch --platform linux --python "^3.7"

---

在这个问题上花了几个小时之后，我找到了一个“解决方案”，将诗歌和pip结合起来，只为PyTorch。您不需要直接指定控制盘URL，因此保持跨平台

我正在使用，一个很好的诗歌任务运行程序，可以运行任意命令

[tool.poetry.dev-dependencies]
poethepoet = "^0.10.0"

[tool.poe.tasks]
force-cuda11 = "python -m pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html"

您可以运行：

poetry install

然后：

poe force-cuda11  # relies on pip and use PyTorch wheels repo

---

这不是一个令人满意的答案，因为这并不能解决以下问题：根据平台的不同，指向正确的方向盘。这是一个很好的问题链接。请注意，另一个相关问题仍然存在，因此这仍然是一个问题，目前没有稳定的解决方案