诗歌与Pythorch

2024-05-01 21:31:28 发布

您现在位置:Python中文网/ 问答频道 /正文

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

似乎没有明显的方法来决定安装哪个Pythorch版本。我考虑过添加不同的安装说明作为extra依赖项,但是我找不到一个选项来获得等效的设置,比如:

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命令。在

我会很感激你的帮助。:)


Tags: 项目httpsorg版本关系downloadhtmltorch