Skip to content

Env modification for 40XX GPU #6

@RichieZou

Description

@RichieZou

Hi, thank you for sharing this powerful lib.

For people who is using 40XX GPU can try this modification for cuda to solve env problem (nvrtc: error: invalid value for --gpu-architecture (-arch)):

pip install torch==1.13.0+cu116 torchaudio==0.13.0+cu116 torchvision==0.14.0+cu116 --index-url https://download.pytorch.org/whl/cu116

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions