1. 显式指定 RTX 5090 的 Blackwell 架构 (sm_120),也可带上通用架构
export TORCH_CUDA_ARCH_LIST="12.0"
export FORCE_CUDA=1
2. 指定单卡并使用 --no-build-isolation 进行安装
CUDA_VISIBLE_DEVICES=0 pip install -e . --no-build-isolation
export TORCH_CUDA_ARCH_LIST="12.0"
export FORCE_CUDA=1
CUDA_VISIBLE_DEVICES=0 pip install -e . --no-build-isolation