8000 [poetry] 2.7.0+cpu includes cuda as a dependency · Issue #152121 · pytorch/pytorch · GitHub
[go: up one dir, main page]

Skip to content
[poetry] 2.7.0+cpu includes cuda as a dependency #152121
@peter-axion

Description

@peter-axion

🐛 Describe the bug

I use torch +cpu variants in images I run on VMs without GPUs because the CUDA libraries are huge, so if I don't need them then I definitely don't want them.

When I use poetry lock on poetry 1 or 2 with torch 2.7.0+cpu in my pyproject.toml, the cuda libraries and triton are added as dependencies (11GB image), while with 2.5.1+cpu, they were not (3.7GB image).

I'm reporting this as a bug because I assume it was unintentional, given that the +cpu addition seems to imply the user won't be running on a GPU.

Versions

I already deleted the big image with torch 2.7.0+cpu :(

cc @ezyang @gchanan @zou3519 @kadeng @msaroufim @seemethere @malfet @osalpekar @atalman

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions

    0