分享价值
提升价值

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

1V1答疑支持
爱发电:点击前往
知识星球:点击前往
赞(0) 打赏
未经允许不得转载:诺瓦小站 » python报错:xFormers can't load C++/CUDA extensions. xFormers was built for: PyTorch 2.1.0+cu121 with CUDA 1201

觉得文章对你有帮助就打赏一下作者

非常感谢你的打赏,我们将继续提供更多优质内容,让我们一起创建更加美好的世界!

支付宝扫一扫

微信扫一扫