To simplify environment setup for Composer, we provide a set of pre-built Docker images.
The mosaicml/composer images contain all Composer pre-installed with
all dependencies for both NLP and Vision models. They are built on top of the
mosaicml/pytorch_vision family of images.
(See the section on MosaicML PyTorch Images below.)
Note: Only the Dockerimage for most recent version of Composer will be maintained. We recommend using
mosaicml/composer:latest or mosaicml/composer:latest_cpu, which will always be up to date.
| Composer Version | CUDA Support | Docker Tag |
|---|---|---|
| 0.13.1 | Yes | mosaicml/composer:latest, mosaicml/composer:0.13.1 |
| 0.13.1 | No | mosaicml/composer:latest_cpu, mosaicml/composer:0.13.1_cpu |
Note: For a lightweight installation, we recommended using a MosaicML PyTorch Image and manually installing Composer within the image.
The mosaicml/pytorch images contain PyTorch preinstalled, without Composer.
The base flavor contains PyTorch pre-installed; the vision flavor also includes OpenCV, MM Segmentation, and FFCV dependencies.
To install composer, once inside the image, run pip install mosaicml.
| Linux Distro | Flavor | PyTorch Version | CUDA Version | Python Version | Docker Tags |
|---|---|---|---|---|---|
| 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 is installed by default in all images.
Pre-built images can be pulled from MosaicML's DockerHub Repository.
For example:
docker pull mosaicml/composer
Content type
Image
Digest
sha256:5960d976a…
Size
7.7 GB
Last updated
about 1 year ago
docker pull mosaicml/pytorch