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人工答疑支持
知识星球(2月2.5折限时优惠中):扫码加入↓ ↓ ↓
赞(0) 打赏
软件无法使用?点击查看常见问题说明>>

最新发布

ToDesk云电脑运行AI软件整合包使用方法-诺瓦小站

ToDesk云电脑运行AI软件整合包使用方法

我分享的大部分AI软件对电脑配置要求都是比较高的。比如显卡一般都要英伟达8G以上显存,有时CPU和内存也有要求,这类配置比较好用的电脑一般都要8000以上,如果你没有电脑或是预算比较充足,可以直接购买高配电脑,如果已经有电脑了或是预算达不到...

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

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

支付宝扫一扫

微信扫一扫