ocpub/oc9-megatron-lm repository overview

⁠Megatron-LM 0.16.0 (GPU) on OpenCloudOS 9

⁠基本信息

• 框架版本：Megatron-Core 0.16.0
• 基础镜像：opencloudos9-cuda-devel:12.8
• Python 版本：3.x（随基础镜像提供）
• CUDA 版本：12.8
• PyTorch 版本：2.11.0

本镜像基于 OpenCloudOS 9 CUDA 开发环境构建，预装：

• PyTorch 2.11.0 (CUDA 12.8)
• torchvision 0.26.0
• torchaudio 2.11.0
• Megatron-Core 0.16.0

适用于：

• Megatron-LM 模型训练
• GPT / LLM 大模型分布式训练
• CUDA GPU 加速推理与实验

⁠构建

docker build -t oc9-megatron-lm:0.16.0 .