训练指南

在 Alauda AI 上对 LLMs 进行微调和预训练的端到端配方。

选择路径

你想要的场景使用指南
可复用模板、可重复运行、可选的 Kueue 配额Kubeflow Trainer v2 + LlamaFactory使用 Kubeflow Trainer v2 进行微调
在崩溃、抢占或暂停后恢复,而不会损失数小时的训练时间Checkpoint PVC + resume_from_checkpoint + maxRestarts检查点保存与恢复 TrainJobs
将训练与在线推理结合,在需要时归还 GPUKueue cohort + 抢占 + 检查点恢复使用 Kueue 的可抢占 TrainJobs
在 GPU 的一个 切片 上运行微调,而不是整卡Dynamic Resource Allocation (DRA) MIG slice + Trainer v2使用 Dynamic Resource Allocation (DRA) 进行 GPU 切片
一组 Curated 的 TrainingRuntime 镜像(CUDA / CANN)Trainer v2 runtime catalogTraining Runtime 镜像
在 Trainer v2 上一键快速开始分布式 PyTorchClusterTrainingRuntime + MNISTKubeflow Trainer 快速开始
具有自动内存管理的生产级 SFT / OSFTtraining_hub使用 Training Hub 微调 LLMs
交互式探索、自定义脚本、VolcanoJob 提交Workbench Notebook使用 Workbench 进行 LLMs 微调
在 Ascend NPU 上进行全参数 SFT / 预训练Workbench PyTorch CANN / MindSpore CANN在 Ascend NPU 上微调和预训练
在 MLflow 中跟踪 KFP 运行的参数和指标KFP component + MLflow SDKKubeflow Pipeline + MLflow 集成
使用可复用的 AutoGluon assets 训练表格或时间序列模型Managed KFP pipelines or composable components使用可复用的 Kubeflow Pipeline 组件
使用 KFP 缓存、并行循环和持久化的 typed artifactsKFP 2.16.1 execution mechanismsKubeflow Pipelines 执行与存储行为
通过 MLflow + TrustyAI 实现每日微调 → 评估 → 比较循环KFP Recurring Run + MLflow Model Registry + LMEvalJob结合 MLflow 和 TrustyAI 的每日微调流水线