
mosaicml/composer为简化Composer的环境设置,我们提供了一系列预构建的Docker镜像。
mosaicml/composer镜像包含预安装的Composer及其所有NLP和视觉模型依赖,构建于mosaicml/pytorch_vision系列镜像之上。(参见下文MosaicML PyTorch镜像部分。)
注意:仅维护最新版本Composer的Docker镜像。建议使用mosaicml/composer:latest或mosaicml/composer:latest_cpu,它们将始终保持最新。
| 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 |
注意:如需轻量级安装,建议使用MosaicML PyTorch镜像并在镜像内手动安装Composer。
mosaicml/pytorch镜像包含预安装的PyTorch,不含Composer。基础版本包含预安装的PyTorch;视觉版本还包括OpenCV、MM Segmentation和FFCV依赖。如需安装Composer,进入镜像后运行pip install mosaicml即可。
| Linux发行版 | 版本类型 | 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 |
所有镜像中默认安装了Pillow-SIMD。
预构建镜像可从MosaicML的DockerHub仓库拉取。
例如:
bashdocker pull mosaicml/composer

manifest unknown 错误
TLS 证书验证失败
DNS 解析超时
410 错误:版本过低
402 错误:流量耗尽
身份认证失败错误
429 限流错误
凭证保存错误
来自真实用户的反馈,见证轩辕镜像的优质服务