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

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐