TrainingSuite 产品介绍
TrainingSuite 是面向摩尔线程 GPU 的训练与推理组件集成镜像。本节介绍 v2.1.0～v2.1.7 的公网镜像、MUSA SDK、PyTorch 组合和适用平台，帮助你在开始训练前选择合适的版本。
1. TrainingSuite 是什么
TrainingSuite 将 Megatron-LM、MT-Slime、MT-Verl、SGLang 和 PyTorch、torch_musa 等组件组合成训练环境。不同版本对应不同的软件栈，不能只根据 TrainingSuite 版本号推断 SDK 或 PyTorch 版本。
1.1 与 PyTorch 基础镜像的区别
| 镜像类型 | 主要内容 | 用途 |
|---|---|---|
| PyTorch 基础镜像 | PyTorch、torch_musa 及其运行依赖 | 开发 PyTorch 程序或作为上层镜像基础 |
| TrainingSuite 镜像 | PyTorch 栈、训练和推理组件及相关适配 | 运行经过集成的训练与推理流程 |
TrainingSuite 镜像虽然包含 PyTorch，但不等同于独立发布的 PyTorch 基础镜像。下表地址均为 TrainingSuite 镜像地址。
2. 版本与公网镜像
| 版本 | 公网镜像 | MUSA SDK / 软件栈 | Python | PyTorch / torch_musa | 适用平台 |
|---|---|---|---|---|---|
| v2.1.7 | registry.mthreads.com/mcctest/training-suite:v2.1.7 | MUSA SDK 4.3.8 | 3.10 | 2.7.1 / 2.7.1 | 常规 x86、Ubuntu 22.04 |
| v2.1.7 Hygon | registry.mthreads.com/mcctest/training-suite:v2.1.7_hygon | MUSA SDK 4.3.8 | 3.10 | 2.7.1 / 2.7.1 | Hygon、Ubuntu 22.04；训练相关仍在适配 |
| v2.1.6 | registry.mthreads.com/mcctest/training-suite:v2.1.6 | MUSA SDK 4.3.5 | 3.10 | 2.7.1 / 2.7.1 | x86_64、Ubuntu 22.04 |
| v2.1.5 | registry.mthreads.com/mcctest/ai/training-suite:v2.1.5 | MUSA SDK 4.3.5 | 3.10 | 2.7.1 / 2.7.1 | x86_64、Ubuntu 22.04 |
| v2.1.4 | registry.mthreads.com/mcctest/ai/training-suite:v2.1.4-4.3.5 | MUSA SDK 4.3.5 | 3.10 | 2.7.1 / 2.7.1 | x86_64、Ubuntu 22.04 |
| v2.1.3 | registry.mthreads.com/mcctest/ai/training-suite:v2.1.3 | MUSA SDK 4.3.5 | 3.10 | 2.7.1 / 2.7.1 | x86_64、Ubuntu 22.04 |
| v2.1.2 | registry.mthreads.com/mcctest/ai/training-suite:v2.1.2 | MUSA SDK 4.3.5 | 3.10 | 2.7.1 / 2.7.1 | x86_64、Ubuntu 22.04 |
| v2.1.1 | registry.mthreads.com/mcctest/ai/training-suite:v2.1.1 | MUSA SDK 4.3.5 | 3.10 | 2.7.1 / 2.7.1 | x86_64、Ubuntu 22.04 |
| v2.1.0 | registry.mthreads.com/mcctest/ai/training-suite:v2.1.0 | MUSA SDK 4.3.5 | 3.10 | 2.7.1 / 2.7.1 | x86_64、Ubuntu 22.04 |
进入容器后应检查实际组件版本。
平台兼容性
v2.1.0～v2.1.5 仅支持 x86_64 架构和 Ubuntu 22.04。v2.1.7 的常规 x86 与 Hygon 镜像不能混用。
3. 如何阅读版本文档
每个版本包含三篇文档：
- 产品介绍：提供统一的产品定位和版本镜像对照。
- 快速开始：说明对应版本的镜像拉取、容器启动和环境检查方法。
- 发布说明：记录对应版本的公网镜像、运行环境、版本更新和支持模型。
初次使用请先阅读快速开始，再查看当前版本的发布说明。
