Skip to content

[ROCm 7.14] Pin PyTorch to 2.11 to avoid high CPU usage #20

Description

@LckHot

Hi, thank you for maintaining this repository.

To avoid the issue reported in ROCm issue #6406, could you please consider pinning the PyTorch version for the ROCm 7.14 environment to PyTorch 2.11 instead of 2.12?

According to the report, PyTorch 2.12 may cause abnormally high CPU usage after the first GPU operation, while downgrading to PyTorch 2.11 resolves the issue.

It may also be necessary to pin compatible versions of torchvision and torchaudio.

Thank you for your consideration.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions