Deepspeed报错解决:Error building extension ‘cpu_adam‘,AttributeError: ‘DeepSpeedCPUAdam‘ object has no a

RuntimeError: Error building extension ‘cpu_adam’
Exception ignored in: <function DeepSpeedCPUAdam.del at 0x7f5b402eb0d0>

AttributeError: ‘DeepSpeedCPUAdam’ object has no attribute ‘ds_opt_adam’

使用huggingface和deepspeed时遇到上述问题。

碰到上面的报错也查了很多资料,但是都没有很好的解决,主要问题还是torch和deepspeed版本的问题。

报错版本:
torch == 2.2.2
deepspeed == 0.14.4

降低deepspeed版本后解决问题:
torch== 2.2.2
deepspeed== 0.14.2

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值