python报错：xFormers can't load C++/CUDA extensions. xFormers was built for: PyTorch 2.1.0+cu121 with CUDA 1201

python应用运行报错，部分信息如下：

WARNING[XFORMERS]: xFormers can’t load C++/CUDA extensions. xFormers was built for:
PyTorch 2.1.0+cu121 with CUDA 1201 (you have 2.1.0+cpu)
Python 3.10.11 (you have 3.10.11)
Please reinstall xformers (see https://github.com/facebookresearch/xformers#installing-xformers)
Memory-efficient attention, SwiGLU, sparse and more won’t be available.
Set XFORMERS_MORE_DETAILS=1 for more details
False

首先xFormers要求安装的是cuda版本的pytorch，我电脑装的是cpu版本的，先卸载掉原来的pytorch:

pip uninstall torch torchaudio torchvision

由于应用要求的是torch 2.1.0版本，我用的是CUDA11.8版本，运行下面命令安装

# CUDA 11.8
pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118

然后再重新安装xformers,先运行下面命令卸载:

pip uninstall xformers

由于安装xformers需要对应正确版本，部分版本信息如下：

xformers (from versions: 0.0.22.post2+cu118, 0.0.22.post3+cu118, 0.0.22.post4+cu118, 0.0.22.post7+cu118, 0.0.23+cu118, 0.0.23.post1+cu118, 0.0.24+cu118, 0.0.25+cu118, 0.0.25.post1+cu118, 0.0.26+cu118, 0.0.26.post1+cu118, 0.0.27+cu118, 0.0.27.post1+cu118, 0.0.27.post2+cu118)

应用要求安装0.0.23以下版本xformers，我选择0.0.22.post7，那么完整安装命令就是：

pip install -U xformers==0.0.22.post7 --index-url https://download.pytorch.org/whl/cu118

如果你cuda版本是12.1或其它，你可以把链接改为https://download.pytorch.org/whl/cu121等

重新运行应用检查，问题解决，应用不再报xformers错误