Skip to content

python3Packages.torchWithRocm: 2.0.0 unnecessarily requires unfree dependencies #224689

@SomeoneSerge

Description

@SomeoneSerge

Issue description

Once we update python3Packages.torch to 2.X.X, its torchWithRocm variation will depend on cudaPacakges.cuda_nvcc (through python3Packages.openai-triton), which is unfree. This means hydra won't be building and caching torchWithRocm.

Upstream enforces that there be a copy of bin/ptxas (from cuda_nvcc):

However, they do not actually use it except for cuda support (obviosuly).
We should patch around this to re-enable caching, and we should work with upstream to make the dependency optional in the first place

CC @NixOS/rocm-maintainers

Metadata

Metadata

Assignees

No one assigned

    Labels

    6.topic: pythonPython is a high-level, general-purpose programming language.
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions