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错误