由于依赖问题，无法在诗歌中安装torch和torchvision的gpu版本

Question

我正在尝试使用诗歌为机器学习创建一个虚拟环境。所以，我使用 pytorch 作为深度学习的框架。我将提取我的 pyproject.toml 的相关部分。

[tool.poetry.dependencies].
python = "^3.8"
torch = { url = "https://download.pytorch.org/whl/cu111/torch-1.8.0%2Bcu111-cp38-cp38-linux_x86_64.whl"}
torchvision = { url = "https://download.pytorch.org/whl/cu111/torchvision-0.9.0%2Bcu111-cp38-cp38-linux_x86_64.whl" }

由于pytroch使用GPU，需要通过指定whl文件来安装。如果以这种方式安装，pytroch 的版本将是1.8.0+cu111。 torchvision对应pytorch的1.8.0就是0.9.0。这个torchvision 所依赖的pytroch 版本是1.8.0（没有cu111）。因此，我无法使用诗歌创建虚拟环境，出现以下错误。

  SolverProblemError

  Because torchvision (0.9.0) depends on torch (1.8.0)
   and mdwithpriorenergy depends on torch (1.8.0+cu111), torchvision is forbidden.
  So, because mdwithpriorenergy depends on torchvision (0.9.0), version solving failed.

So, because [env name] depends on torchvision (0.9.0), version solving failed. 

我还对上面pyproject.toml中的torchvision进行了以下更改，但没有奏效。

[tool.poetry.dependencies].
python = "^3.8"
torch = { url = "https://download.pytorch.org/whl/cu111/torch-1.8.0%2Bcu111-cp38-cp38-linux_x86_64.whl"}
- torchvision = { url = "https://download.pytorch.org/whl/cu111/torchvision-0.9.0%2Bcu111-cp38-cp38-linux_x86_64.whl"}
+ torchvision = "*"

在这种情况下，我收到以下错误

  AttributeError

  'EmptyConstraint' object has no attribute 'allows'.

请告诉我如何解决这个错误。

Answer 1

据我所知，这在诗歌中尚不支持（没有丑陋的黑客），请参阅Issue 2613。

也就是说，我正在维护一个名为 relaxed-poetry 的分支。这是一个非常年轻的分支，但它通过以下配置支持你想要的：

[tool.poetry.dependencies]
python = "^3.8"
torch = { version = "=1.8.0+cu111", source = "pytorch" }
torchvision = { version = "=0.9.0+cu111", source = "pytorch" }

[[tool.poetry.source]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cu111/"
secondary = true

如果你觉得它对你有帮助，你可以把它和诗歌并排安装，改用命令rp

请注意，安装将需要一些时间，因为这是一个很大的依赖项和相对较慢的源..