mosaicml/pytorch_vision
mosaicml
MosaicML提供的预构建Docker镜像，包括预装Composer及NLP/Vision依赖的composer镜像，和预装PyTorch的pytorch镜像（分Base和Vision flavor），用于简化深度学习环境配置。
1 次收藏下载次数: 0状态：社区镜像维护者：mosaicml仓库类型：镜像最近更新：2 年前
让 AI 帮你使用轩辕镜像？ · 展开查看说明 · 点击收起说明
如果你使用 DeepSeek、元宝 AI、Copilot、ChatGPT 等 AI 工具，协助配置轩辕镜像、编写 docker pull 命令、修改 Docker Compose 镜像地址、配置镜像加速、排查镜像拉取失败、分析报错日志等问题，请先让 AI 阅读并遵守轩辕镜像的规则文档。
只需在 AI 对话中先发送下面这句话即可：
请先完整阅读并严格遵守以下文档中的全部规则与要求：
https://xuanyuan.cloud/agents.md
在未充分阅读并理解该文档前，不要生成任何命令、配置、修改建议、故障排查方案或技术回答。后续所有输出都必须严格以该文档中的规范为最高优先级执行。查看 agents.md 用法指南与完整示范。国内用户首推 元宝 AI、DeepSeek 的深度思考模式，不推荐豆包 AI；Cursor 等编辑器可在对话 @ 该链接，或加入 User Rules。 若 AI 无法访问外链，可 打开说明文档 复制全文粘贴。文档会随站点更新，复制内容可能过期，建议定期检查。
MosaicML Docker镜像文档
镜像概述和主要用途
MosaicML提供一系列预构建Docker镜像，旨在简化深度学习环境的配置过程。这些镜像主要分为两类：mosaicml/composer镜像和mosaicml/pytorch镜像。composer镜像预装了Composer及其所有NLP和Vision模型依赖，而pytorch镜像则仅预装PyTorch，适用于需要轻量级安装的场景。
核心功能和特性
Composer镜像
- 基于
mosaicml/pytorch_vision系列镜像构建 - 预装Composer及所有NLP和Vision模型依赖
- 提供支持CUDA和CPU的版本
- 仅维护最新版本，推荐使用
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 |
PyTorch镜像
- 预装PyTorch，不含Composer
- 提供Base和Vision两种flavor：
- Base：仅包含PyTorch预安装
- Vision：额外包含OpenCV、MM Segmentation和FFCV依赖
- 支持多种PyTorch版本、CUDA版本和Python版本
- 可在镜像内通过
pip install mosaicml手动安装Composer - 所有镜像默认安装
Pillow-SIMD
PyTorch镜像版本矩阵
| 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 |
使用场景和适用范围
- Composer镜像：适用于需要快速启动Composer环境，进行NLP或Vision模型开发、训练的场景。
- PyTorch镜像：适用于需要轻量级PyTorch环境，或希望自定义安装Composer及其他依赖的场景。
- Vision flavor：特别适合计算机视觉任务，已包含OpenCV等常用视觉库。
使用方法和配置说明
拉取镜像
预构建镜像可从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
若使用PyTorch镜像并需安装Composer，可在容器内执行以下命令：
bashpip install mosaicml
镜像拉取方式
您可以使用以下命令拉取该镜像。请将 <标签> 替换为具体的标签版本。如需查看所有可用标签版本，请访问 标签列表页面。
轩辕镜像加速拉取命令点我查看更多 pytorch_vision 镜像标签
docker pull docker.xuanyuan.run/mosaicml/pytorch_vision:<标签>
DockerHub 原生拉取命令
docker pull mosaicml/pytorch_vision:<标签>
镜像拉取常见问题
功能
错误码
用户好评
来自真实用户的反馈，见证轩辕镜像的优质服务
oldzhang
运维工程师
Linux服务器
5
"Docker访问体验非常流畅，大镜像也能快速完成下载。"