
MosaicML提供一系列预构建Docker镜像,旨在简化深度学习环境的配置过程。这些镜像主要分为两类:mosaicml/composer镜像和mosaicml/pytorch镜像。composer镜像预装了Composer及其所有NLP和Vision模型依赖,而pytorch镜像则仅预装PyTorch,适用于需要轻量级安装的场景。
mosaicml/pytorch_vision系列镜像构建latest或latest_cpu标签确保更新Composer镜像版本矩阵
| Composer版本 | CUDA支持 | Docker标签 |
|---|---|---|
| 0.13.1 | 是 | mosaicml/composer:latest, mosaicml/composer:0.13.1 |
| 0.13.1 | 否 | mosaicml/composer:latest_cpu, mosaicml/composer:0.13.1_cpu |
pip install mosaicml手动安装ComposerPillow-SIMDPyTorch镜像版本矩阵
| Linux发行版 | Flavor | PyTorch版本 | CUDA版本 | Python版本 | Docker标签 |
|---|---|---|---|---|---|
| Ubuntu 20.04 | Base | 1.13.1 | 11.7.1 (Infiniband) | 3.10 | mosaicml/pytorch:latest, mosaicml/pytorch:1.13.1_cu117-python3.10-ubuntu20.04 |
| Ubuntu 20.04 | Base | 1.13.1 | 11.7.1 (EFA) | 3.10 | mosaicml/pytorch:latest-aws, mosaicml/pytorch:1.13.1_cu117-python3.10-ubuntu20.04-aws |
| Ubuntu 20.04 | Base | 1.13.1 | cpu | 3.10 | mosaicml/pytorch:latest_cpu, mosaicml/pytorch:1.13.1_cpu-python3.10-ubuntu20.04 |
| Ubuntu 20.04 | Base | 1.12.1 | 11.6.2 (Infiniband) | 3.9 | mosaicml/pytorch:1.12.1_cu116-python3.9-ubuntu20.04 |
| Ubuntu 20.04 | Base | 1.12.1 | cpu | 3.9 | mosaicml/pytorch:1.12.1_cpu-python3.9-ubuntu20.04 |
| Ubuntu 20.04 | Base | 1.11.0 | 11.5.2 (Infiniband) | 3.8 | mosaicml/pytorch:1.11.0_cu115-python3.8-ubuntu20.04 |
| Ubuntu 20.04 | Base | 1.11.0 | cpu | 3.8 | mosaicml/pytorch:1.11.0_cpu-python3.8-ubuntu20.04 |
| Ubuntu 20.04 | Vision | 1.13.1 | 11.7.1 (Infiniband) | 3.10 | mosaicml/pytorch_vision:latest, mosaicml/pytorch_vision:1.13.1_cu117-python3.10-ubuntu20.04 |
| Ubuntu 20.04 | Vision | 1.13.1 | cpu | 3.10 | mosaicml/pytorch_vision:latest_cpu, mosaicml/pytorch_vision:1.13.1_cpu-python3.10-ubuntu20.04 |
预构建镜像可从https://hub.docker.com/u/mosaicml%E6%8B%89%E5%8F%96%E3%80%82
拉取Composer镜像示例
bashdocker pull mosaicml/composer
拉取PyTorch镜像示例
bash# 拉取最新PyTorch Base镜像(带CUDA) docker pull mosaicml/pytorch:latest # 拉取最新PyTorch CPU镜像 docker pull mosaicml/pytorch:latest_cpu # 拉取PyTorch Vision镜像 docker pull mosaicml/pytorch_vision:latest
若使用PyTorch镜像并需安装Composer,可在容器内执行以下命令:
bashpip install mosaicml
您可以使用以下命令拉取该镜像。请将 <标签> 替换为具体的标签版本。如需查看所有可用标签版本,请访问 标签列表页面。






探索更多轩辕镜像的使用方法,找到最适合您系统的配置方式
通过 Docker 登录认证访问私有仓库
无需登录使用专属域名
Kubernetes 集群配置 Containerd
K3s 轻量级 Kubernetes 镜像加速
VS Code Dev Containers 配置
Podman 容器引擎配置
HPC 科学计算容器配置
ghcr、Quay、nvcr 等镜像仓库
Harbor Proxy Repository 对接专属域名
Portainer Registries 加速拉取
Nexus3 Docker Proxy 内网缓存
需要其他帮助?请查看我们的 常见问题Docker 镜像访问常见问题解答 或 提交工单
docker search 限制
站内搜不到镜像
离线 save/load
插件要用 plugin install
WSL 拉取慢
安全与 digest
新手拉取配置
镜像合规机制
manifest unknown
no matching manifest(架构)
invalid tar header(解压)
TLS 证书失败
DNS 超时
域名连通性排查
410 Gone 排查
402 与流量用尽
401 认证失败
429 限流
D-Bus 凭证提示
413 与超大单层
来自真实用户的反馈,见证轩辕镜像的优质服务