[Docs] What GPU versions are supported by pre-built pip wheels in PyPI?

[samskalicky]

📚 The doc issue

Hello,

I would like to know what CUDA arch versions (ie. sm_*) are used to build the pre-built pip wheels installed from PyPI. Here are the GPU arch im interested in using MMCV with:

Nvidia T4 - AWS EC2 G4 - sm_75
Nvidia A10G - AWS EC2 G5 - sm_86
Nvidia A100 - AWS EC2 P4 - sm_80
Nvidia H100 - AWS EC2 P5 - sm_90

I found this usage for CI:
https://github.com/open-mmlab/mmcv/blob/main/.github/workflows/merge_stage_test.yml#L257

But where are the GPU arch set when building the pip wheels that are published to PyPI? Can you point to a script or build file?

Thank you!
Sam

Suggest a potential alternative/fix

No response