AI软件下载
有趣网站推荐及实用软件下载

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

电脑不给力运行不了?试试高配云电脑>>
如需1对1人工答疑支持
爱发电:点击前往
赞(0) 打赏
软件无法使用?点击查看常见问题说明>>

最新发布

ToDesk、青椒云、顺网云、海马云电脑使用体验分享-诺瓦小站

ToDesk、青椒云、顺网云、海马云电脑使用体验分享

最近调试了一些大的AI应用,我那渣渣电脑带不动,又不适合买新电脑,所以网上找了一些高端显卡云电脑作为替代,最近用了几家,如:ToDesk云电脑、青椒云电脑、顺网云电脑、海马云电脑,各有优劣吧,这里总结分享一下使用体验。 云电脑对比分析 海马...

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

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

支付宝扫一扫

微信扫一扫