分享价值
提升价值

xFormers can't load C++/CUDA extensions. xFormers was built for:PyTorch 2.2.2+cu118 with CUDA 1108 (you have 2.2.2+cu121)解决方法

python应用运行报错,部分输出信息如下:WARNING[XFORMERS]: xFormers can’t load C++/CUDA extensions. xFormers was built for:
PyTorch 2.2.2+cu118 with CUDA 1108 (you have 2.2.2+cu121)
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

pip list查看安装的依赖信息,可以看到torch 版本是2.2.2+cu121,xformers版本是0.0.25.post1+cu118,可以看到cuda版本不一样,可以把xformers卸载重装,先

pip uninstall xformers

查看requirements.txt内容xformers要求是0.0.25.post1版本,可以运行如下命令重新安装cuda121版本的xformers:

pip install -U xformers==0.0.25.post1 --index-url https://download.pytorch.org/whl/cu121

成功安装,重新运行没有再报xFormers错误。

xformers/CUDA/pytorch版本关系对照参考文章:《最新xformers/CUDA/pytorch版本关系对照表

1V1答疑支持
爱发电:点击前往
知识星球:点击前往
赞(0) 打赏
未经允许不得转载:诺瓦小站 » xFormers can't load C++/CUDA extensions. xFormers was built for:PyTorch 2.2.2+cu118 with CUDA 1108 (you have 2.2.2+cu121)解决方法

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

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

支付宝扫一扫

微信扫一扫