swift微调大模型遇到的问题及解决方案

微调时执行finetune.sh文件后报错:

        ImportError: /root/.cache/torch_extensions/py312_cu118/fused_adam/fused_adam.so: cannot open shared 

问题原因:

        在安装deepspeed时直接使用了以下命令,导致安装时没有对fused_adam进行安装;

pip install deepspeed

        通过在命令行输入以下命令,可查看到deepspeed的具体安装信息;

 ds_report

        具体显示如下(图上传不了,简单描述以下)

        主要看到fused_adam那一行,其install的信息栏显示为NO

解决办法:

       先卸载deepspeed然后通过源码下载安装:

pip uninstall deepspeed
git clone https://github.com/microsoft/DeepSpeed
cd DeepSpeed
DS_BUILD_FUSED_ADAM=1 pip install .

 再次执行训练,发现已经不在报上述错误。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值