首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >由于依赖问题,无法以诗歌形式安装torch和torchvision的gpu版本

由于依赖问题,无法以诗歌形式安装torch和torchvision的gpu版本
EN

Stack Overflow用户
提问于 2021-08-23 12:24:31
回答 1查看 144关注 0票数 1

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

代码语言:javascript
运行
复制
[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。pytorch的1.8.0对应的torchvision0.9.0。此torchvision所依赖的pytroch版本是1.8.0 (没有cu111)。因此,我不能使用poetry创建虚拟环境,错误如下。

代码语言:javascript
运行
复制
  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做了以下更改,但它们不起作用。

代码语言:javascript
运行
复制
[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 = "*"

在本例中,我收到以下错误

代码语言:javascript
运行
复制
  AttributeError

  'EmptyConstraint' object has no attribute 'allows'.

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

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-08-29 17:37:05

据我所知,这在诗歌中还不受支持(没有丑陋的技巧),请参阅Issue 2613

也就是说,我正在维护一个叫做relaxed-poetry的分支。它是一个非常年轻的分支,但它通过以下配置支持您想要的东西:

代码语言:javascript
运行
复制
[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

如果您认为它可以帮助您,您可以将其与poetry并排安装,并改用命令rp

但请注意,安装将需要一些时间,因为这是一个很大的依赖项,并且是一个相对较慢的源代码。

票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/68892660

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档