xFormers无法加载C++/CUDA扩展及PyTorch依赖冲突求助

问题原因 #

1. xFormers编译版本与当前PyTorch不兼容：原xFormers是为PyTorch 2.1.0+cu121构建，当前环境为PyTorch 2.2.1+cu121，导致无法加载CUDA扩展
2. 直接升级xFormers时，pip自动将PyTorch降级到2.2.0，与已安装的torchaudio、torchtext、torchvision（均要求PyTorch 2.2.1）产生依赖冲突

解决步骤 #

1. 统一PyTorch全家桶版本到2.2.1+cu121
   执行以下命令，确保torch、torchaudio、torchvision版本完全匹配：

   !pip install torch==2.2.1+cu121 torchaudio==2.2.1+cu121 torchvision==0.17.1+cu121 --index-url https://download.pytorch.org/whl/cu121
   

2. 重新安装适配PyTorch 2.2.1+cu121的xFormers
   完成第一步后，执行命令安装对应版本的xFormers：

   !pip install -U xformers --index-url https://download.pytorch.org/whl/cu121
   

3. 验证修复效果
   重启Colab运行时后，重新启动Fast SD Automatic1111，检查xFormers加载警告是否消失

内容的提问来源于stack exchange，提问作者supern