我最近发现诗歌可以帮助解决我们小组的所有不同依赖问题。在一个项目中,我们也使用PyTorch,我想知道是否有人可以帮助我并指出如何将其添加到诗歌中的方向。我们正在研究无法访问 CUDA GPU(用于简单的道路推理/测试)的机器和我们可以访问 CUDA CPU 的工作站。现在我很想用诗歌来确保每个开发者都在诗歌的帮助下使用相同的版本,但这可能吗?
似乎没有明显的方法来决定安装哪个 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
但我宁愿不直接在 git 中使用它们......我在诗歌中看到的最接近的选项是手动下载它们然后使用file = X
命令。
我将不胜感激任何帮助。:)