MosaicML Docker镜像文档

镜像概述和主要用途

MosaicML提供一系列预构建Docker镜像，旨在简化环境设置。这些镜像分为两类：Composer镜像和PyTorch镜像。Composer镜像预装了Composer及其所有NLP和Vision模型依赖；PyTorch镜像仅预装PyTorch，不含Composer，适合需要轻量级安装的场景。

Composer镜像

mosaicml/composer镜像包含预安装的Composer及其所有NLP和Vision模型依赖，基于mosaicml/pytorch_vision系列镜像构建。

维护策略

注意：仅维护最新版本的Composer Docker镜像。建议使用mosaicml/composer:latest或mosaicml/composer:latest_cpu标签，这些标签将始终保持最新。

Composer镜像版本矩阵

注意：如需轻量级安装，建议使用MosaicML PyTorch镜像并在镜像内手动安装Composer。

PyTorch镜像

mosaicml/pytorch镜像包含预安装的PyTorch，但不含Composer。基础版本（Base）仅预装PyTorch；视觉版本（Vision）还包含OpenCV、MM Segmentation和FFCV依赖。如需安装Composer，可在镜像内运行pip install mosaicml。

PyTorch镜像版本矩阵

所有镜像默认安装Pillow-SIMD。

拉取镜像

预构建镜像可从MosaicML的DockerHub仓库拉取。

例如：

bash
docker pull mosaicml/composer