首先可以在Files | Anaconda.org上查找符合的xformers版本

直接pip安装xformers会默认使用支持该xformers的cuda最高版本,可能和环境的cuda版本不对应

按照安装xFormers时遇到的问题,以及正确的安装方式_xformers安装-CSDN博客中的idex方法可以指定cuda版本:

比如:在cu11.8下安装0.0.26版本的xformers则可以用pip3 install xformers==0.0.26 --index-url https://download.pytorch.org/whl/cu118

问题解决。

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐