诗歌与PyTorch

Question

我最近发现poetry可以管理依赖项。在一个项目中，我们使用PyTorch。如何将其添加到poetry中

我们工作的机器，没有访问CUDA GPU (简单的道路上推断/测试)和工作站，我们确实可以访问CUDA图形处理器。是否可以使用诗歌来确保每个开发人员都使用相同的PyTorch版本？

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

pip3 install torch==1.3.1+cpu torchvision==0.4.2+cpu -f https://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

但我不想让他们直接穿.我在诗歌中看到的最接近的选择是手动下载它们，然后使用file = X命令。

Answer 1

目前，诗歌没有-f选项(有公开发行和开放PR)，所以不能使用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" }

Answer 2

在这个问题上花了几个小时之后，我找到了一个“解决方案”，把“诗歌”和“皮普”结合起来，只为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

Answer 3

本期在诗歌集散地中的最新解决方案：

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