Feature(build): enhance CUDA version handling and architecture detection#144
Feature(build): enhance CUDA version handling and architecture detection#144johnnynunez wants to merge 2 commits intomicrosoft:mainfrom
Conversation
|
Hi @johnnynunez, thanks for your help in extending support to Blackwell and ARM architectures. However, due to two reasons: 1) We currently don't have access to the Blackwell hardware to properly verify correctness, and we’re unable to immediately review and merge this PR. |
No problems. You can check it if you want when you have access: https://pypi.jetson-ai-lab.dev/sbsa/cu128 |
Support aarch64 including support GH200/GB200
Support Blackwell RTX50/A6000 and B100/B200
Support pytorch 2.7.0
Support cuda 12.8.1