Set XFORMERS_MORE_DETAILS=1 for more details

AI算法网奇

已于 2024-02-04 04:41:34 修改

阅读量806

点赞数 10

分类专栏： python基础文章标签： python

于 2024-01-05 14:39:40 首次发布

本文链接：https://blog.csdn.net/jacke121/article/details/135409284

版权

python基础专栏收录该内容

449 篇文章 54 订阅

订阅专栏

文章报道了在尝试使用xformers库时遇到的兼容性问题，提示用户需针对特定CUDA版本（如cu118和cu121）重新安装xformers，以获取内存高效注意力、SwiGLU等特性。

摘要由CSDN通过智能技术生成

报警：

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

报错代码：

import xformers

报错：

WARNING[XFORMERS]: xFormers can't load C++/CUDA extensions. xFormers was built for:
PyTorch 2.2.0+cu118 with CUDA 1108 (you have 2.1.0+cu118)
Python 3.10.11 (you have 3.10.13)
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

解决方法：

# cuda 11.8 version
pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu118
# cuda 12.1 version
pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu121

过程中会重新安装cuda