安装xFormers时遇到的问题,以及正确的安装方式

在使用pip install xformers安装xformers时,发现总是会把我环境中的pytorch重新安装,并且会安装CUDA12版本的pytorch, 而我环境是CUDA 11.8,这就导致原本的开发环境不可用了。后来发现xformers与pytorch版本一一对应的,在pip install xformers时,如果发现pytorch版本不一致,就会自动卸载重装pytorch, 而默认装的是CUDA12版本的。

1. 如果你环境中的pytorch是最新的v2.3.0,那么可以根据你的cuda版本采用以下方式安装最新的xformers

# 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

2. 如果你的pytorch版本低于v2.3.0,则需要相应的降低xformers版本,到底哪个版本适合你的pytorch,可以用如下方式进行查看:

pip3 install xformers==0.0.26.post1 --dry-run
pip3 install xformers==0.0.25 --dry-run
...
pip3 install xformers==0.0.20 --dry-run
pip3 install xformers==0.0.19 --dry-run

最新版本是v0.0.26.post1,依次向下使用上面的命令进行尝试,并不断降低版本,直到显示当前版本所需要的pytorch版本已存在的时候,说明该版本的xformers是你可以安装的。
下面有一些pytorch和xformers依赖版本的对应关系:

xformers版本与其依赖pytorch版本的对应关系

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值