Skip to content

APEX GroupNorm Cannot Be Built with CUDA 12.6 Due to Presence of SM100/120 in setup.py #1919

@nWEIdia

Description

@nWEIdia

Setup.py introduced sm100/sm120 regardless of cuda version being used.
When a user is using cuda 12.6 still, they would encounter failures due to the presence of sm100/sm120.

"nvcc fatal : Unsupported gpu architecture 'compute_100'"

Creating this issue to track the fix.

@ptrblck pointed out that https://github.com/NVIDIA/apex/blame/master/setup.py#L534 needs to be indeed improved. cc @crcrpar

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions