diff --git a/composer/_version.py b/composer/_version.py index 54887b79cb..600bc4d182 100644 --- a/composer/_version.py +++ b/composer/_version.py @@ -3,4 +3,4 @@ """The Composer Version.""" -__version__ = '0.18.0' +__version__ = '0.18.1' diff --git a/docker/README.md b/docker/README.md index e3bab86b5c..32f6f6e0e9 100644 --- a/docker/README.md +++ b/docker/README.md @@ -15,8 +15,8 @@ all dependencies for both NLP and Vision models. They are built on top of the | Composer Version | CUDA Support | Docker Tag | |--------------------|----------------|----------------------------------------------------------------| -| 0.18.0 | Yes | `mosaicml/composer:latest`, `mosaicml/composer:0.18.0` | -| 0.18.0 | No | `mosaicml/composer:latest_cpu`, `mosaicml/composer:0.18.0_cpu` | +| 0.18.1 | Yes | `mosaicml/composer:latest`, `mosaicml/composer:0.18.1` | +| 0.18.1 | No | `mosaicml/composer:latest_cpu`, `mosaicml/composer:0.18.1_cpu` | **Note**: For a lightweight installation, we recommended using a [MosaicML PyTorch Image](#pytorch-images) and manually diff --git a/docker/build_matrix.yaml b/docker/build_matrix.yaml index 7a87276336..7ba413a4bb 100644 --- a/docker/build_matrix.yaml +++ b/docker/build_matrix.yaml @@ -193,9 +193,9 @@ TORCHVISION_VERSION: 0.18.0 - AWS_OFI_NCCL_VERSION: '' BASE_IMAGE: nvidia/cuda:12.1.0-cudnn8-devel-ubuntu20.04 - COMPOSER_INSTALL_COMMAND: mosaicml[all]==0.18.0 + COMPOSER_INSTALL_COMMAND: mosaicml[all]==0.18.1 CUDA_VERSION: 12.1.0 - IMAGE_NAME: composer-0-18-0 + IMAGE_NAME: composer-0-18-1 MOFED_VERSION: 5.5-1.0.3.2 NVIDIA_REQUIRE_CUDA_OVERRIDE: cuda>=12.1 brand=tesla,driver>=450,driver<451 brand=tesla,driver>=470,driver<471 brand=unknown,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471 brand=nvidiartx,driver>=470,driver<471 @@ -216,15 +216,15 @@ PYTORCH_NIGHTLY_VERSION: '' PYTORCH_VERSION: 2.1.2 TAGS: - - mosaicml/composer:0.18.0 + - mosaicml/composer:0.18.1 - mosaicml/composer:latest TARGET: composer_stage TORCHVISION_VERSION: 0.16.2 - AWS_OFI_NCCL_VERSION: '' BASE_IMAGE: ubuntu:20.04 - COMPOSER_INSTALL_COMMAND: mosaicml[all]==0.18.0 + COMPOSER_INSTALL_COMMAND: mosaicml[all]==0.18.1 CUDA_VERSION: '' - IMAGE_NAME: composer-0-18-0-cpu + IMAGE_NAME: composer-0-18-1-cpu MOFED_VERSION: 5.5-1.0.3.2 NVIDIA_REQUIRE_CUDA_OVERRIDE: '' PYTHON_VERSION: '3.10' @@ -232,7 +232,7 @@ PYTORCH_NIGHTLY_VERSION: '' PYTORCH_VERSION: 2.1.2 TAGS: - - mosaicml/composer:0.18.0_cpu + - mosaicml/composer:0.18.1_cpu - mosaicml/composer:latest_cpu TARGET: composer_stage TORCHVISION_VERSION: 0.16.2 diff --git a/docker/generate_build_matrix.py b/docker/generate_build_matrix.py index 28d52aefa9..e51662729d 100644 --- a/docker/generate_build_matrix.py +++ b/docker/generate_build_matrix.py @@ -246,7 +246,7 @@ def _main(): composer_entries = [] # The `GIT_COMMIT` is a placeholder and Jenkins will substitute it with the actual git commit for the `composer_staging` images - composer_versions = ['0.18.0'] # Only build images for the latest composer version + composer_versions = ['0.18.1'] # Only build images for the latest composer version composer_python_versions = [LATEST_PYTHON_VERSION] # just build composer against the latest for product in itertools.product(composer_python_versions, composer_versions, cuda_options):