引子
由于工作上需要,一直在用Qwen做大模型推理,有个再训练的需求,特此琢磨下Qwen的训练。OK,我们开始吧。
一、安装环境
查看显卡驱动版本
根据官网推荐
OK,docker在手,天下我有。
docker pull qwenllm/qwen:cu117
docker run -it --rm --gpus=all -v /mnt/code/LLM_Service/:/workspace qwenllm/qwen:cu117 bash
二、测试环境
1、数据集准备
2、下载代码
cd /workspace/qwen1.5_train/Qwen1.5/examples/sft
3、配置
修改脚本,/workspace/qwen1.5_train/Qwen1.5/examples/sft/finetune.sh
执行命令, bash finetune.sh,报错如下:
修改如下:
报错,显卡不支持bf16,改为fp16精度
本地训练,修改脚本
继续报错,transformer没更新
git install transformer -i Simple Index
执行命令, bash finetune.sh
继续报错,accelerate版本不对
pip install accelerate==0.27.2
单机多卡,继续报错。
修改为单机单卡,重启容器,docker run -it --rm --gpus='"device=1"' -v /mnt/code/LLM_Service/:/workspace qwen:v1.0 bash
out of memory,修改为7B模型重新尝试,下载地址https://huggingface.co/Qwen/CodeQwen1.5-7B-Chat/tree/main
两条数据训练完成
显存占用
标签:--,微调,Qwen,报错,Qwen1.5,workspace,bash From: https://www.cnblogs.com/nick-algorithmer/p/18152540