LLaMA-Factory教程：使用Docker快速部署微调环境

LLaMA-Factory教程使用Docker快速部署微调环境【免费下载链接】LlamaFactoryUnified Efficient Fine-Tuning of 100 LLMs VLMs (ACL 2024)项目地址: https://gitcode.com/GitHub_Trending/ll/LlamaFactory在AI大模型应用开发中环境配置往往是开发者面临的第一道障碍。不同操作系统、依赖版本冲突、硬件加速配置等问题常常耗费大量时间。LLaMA-Factory提供了完善的Docker部署方案让你无需复杂配置即可快速搭建专业的大模型微调环境。本文将以NVIDIA GPU环境为例带你一步到位完成Docker化部署轻松开启大模型微调之旅。环境准备系统要求LLaMA-Factory的Docker部署方案目前仅支持Linux操作系统需确保你的系统满足以下条件已安装Docker引擎推荐20.10以上版本已安装Docker Compose工具用于简化容器管理NVIDIA显卡驱动已正确安装建议525.60.13以上版本已安装NVIDIA Container Toolkit实现Docker容器GPU访问安装依赖组件首先通过以下命令安装Docker和Docker Compose# Ubuntu/Debian系统 sudo apt-get update sudo apt-get install -y docker.io docker-compose接着安装NVIDIA Container Toolkit以启用GPU支持# 添加NVIDIA Docker仓库 distribution$(. /etc/os-release;echo $ID$VERSION_ID) curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list # 安装nvidia-container-toolkit sudo apt-get update sudo apt-get install -y nvidia-container-toolkit # 重启Docker服务 sudo systemctl restart docker验证GPU访问能力安装完成后运行以下命令验证Docker是否能正常访问GPUsudo docker run --rm --gpus all nvidia/cuda:12.4.0-base-ubuntu22.04 nvidia-smi若成功显示GPU信息类似下图则说明环境准备就绪----------------------------------------------------------------------------- | NVIDIA-SMI 535.104.05 Driver Version: 535.104.05 CUDA Version: 12.2 | |--------------------------------------------------------------------------- | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. | | | | MIG M. | || | 0 NVIDIA GeForce ... Off | 00000000:01:00.0 Off | N/A | | 30% 35C P8 11W / 320W | 0MiB / 16384MiB | 0% Default | | | | N/A | ---------------------------------------------------------------------------快速部署步骤获取项目代码首先克隆LLaMA-Factory项目仓库git clone https://gitcode.com/GitHub_Trending/ll/LLaMA-Factory cd LLaMA-Factory使用Docker Compose部署推荐LLaMA-Factory提供了便捷的Docker Compose配置文件位于docker/docker-cuda/docker-compose.yml通过以下命令即可一键启动cd docker/docker-cuda/ docker compose up -d docker compose exec llamafactory bash上述命令会自动完成构建包含CUDA支持的Docker镜像创建并启动容器后台运行模式进入容器内部命令行环境手动构建与运行高级选项如果需要自定义构建参数可以使用docker build命令手动构建镜像# 构建Docker镜像 docker build -f ./docker/docker-cuda/Dockerfile \ --build-arg PIP_INDEXhttps://pypi.org/simple \ --build-arg EXTRASmetrics \ -t llamafactory:latest . # 运行容器 docker run -dit --ipchost --gpusall \ -p 7860:7860 \ -p 8000:8000 \ --name llamafactory \ llamafactory:latest # 进入容器 docker exec -it llamafactory bash其中关键参数说明--gpusall允许容器使用所有GPU设备-p 7860:7860映射WebUI端口-p 8000:8000映射API服务端口--ipchost使用主机的IPC命名空间提升性能验证部署结果容器启动后可以通过以下方式验证环境是否正常工作检查WebUI访问打开浏览器访问http://localhost:7860如果看到LLaMA-Factory的Web界面类似下图说明部署成功运行测试微调在容器内部执行以下命令运行一个简单的LoRA微调测试python src/train.py \ --model_name_or_path meta-llama/Llama-2-7b-hf \ --do_train \ --dataset alpaca_zh_demo \ --finetuning_type lora \ --output_dir ./saved_models/llama2-7b-lora \ --per_device_train_batch_size 4 \ --gradient_accumulation_steps 4 \ --learning_rate 2e-4 \ --num_train_epochs 3 \ --logging_steps 10 \ --save_steps 100 \ --fp16如果训练正常开始并显示类似以下输出则表明微调环境已完全就绪{loss: 1.8243, learning_rate: 0.0002, epoch: 0.15} {loss: 1.6521, learning_rate: 0.000199943661971831, epoch: 0.3} {loss: 1.5874, learning_rate: 0.00019977464788732395, epoch: 0.45}常见问题解决GPU未被检测到如果容器内无法识别GPU请按以下步骤排查确认宿主机nvidia-smi命令能正常显示GPU检查nvidia-container-toolkit是否正确安装dpkg -l | grep nvidia-container-toolkit重启Docker服务sudo systemctl restart docker验证Docker GPU支持docker run --rm --gpus all ubuntu nvidia-smi权限问题若执行Docker命令时出现Permission denied错误将当前用户添加到docker用户组sudo usermod -aG docker $USER # 注销并重新登录使更改生效端口冲突如果7860或8000端口已被占用可以修改docker-compose.yml中的端口映射配置例如ports: - 7861:7860 # 将容器7860端口映射到主机7861端口 - 8001:8000 # 将容器8000端口映射到主机8001端口总结与后续步骤通过Docker部署LLaMA-Factory你已成功搭建起专业的大模型微调环境。接下来可以探索examples/train_lora/目录下的微调配置示例查看README_zh.md了解更多微调方法和参数说明尝试使用WebUI或API接口进行模型微调与推理LLaMA-Factory的Docker方案不仅解决了环境依赖问题还保证了不同机器上的一致性体验让你可以专注于模型调优而非环境配置。立即开始你的大模型微调之旅吧提示定期执行git pull和docker compose up -d --build可以获取最新版本的LLaMA-Factory和Docker镜像。收藏本文档随时查阅部署细节祝你在大模型微调之路上取得成功【免费下载链接】LlamaFactoryUnified Efficient Fine-Tuning of 100 LLMs VLMs (ACL 2024)项目地址: https://gitcode.com/GitHub_Trending/ll/LlamaFactory创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考