diff --git a/composer/_version.py b/composer/_version.py index a41361e246..54887b79cb 100644 --- a/composer/_version.py +++ b/composer/_version.py @@ -3,4 +3,4 @@ """The Composer Version.""" -__version__ = '0.17.2' +__version__ = '0.18.0' diff --git a/docker/README.md b/docker/README.md index 523982f8dd..e3bab86b5c 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.17.2 | Yes | `mosaicml/composer:latest`, `mosaicml/composer:0.17.2` | -| 0.17.2 | No | `mosaicml/composer:latest_cpu`, `mosaicml/composer:0.17.2_cpu` | +| 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` | **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 845ea360c7..7a87276336 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.17.2 + COMPOSER_INSTALL_COMMAND: mosaicml[all]==0.18.0 CUDA_VERSION: 12.1.0 - IMAGE_NAME: composer-0-17-2 + IMAGE_NAME: composer-0-18-0 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.17.2 + - mosaicml/composer:0.18.0 - 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.17.2 + COMPOSER_INSTALL_COMMAND: mosaicml[all]==0.18.0 CUDA_VERSION: '' - IMAGE_NAME: composer-0-17-2-cpu + IMAGE_NAME: composer-0-18-0-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.17.2_cpu + - mosaicml/composer:0.18.0_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 b8aa136109..28d52aefa9 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.17.2'] # Only build images for the latest composer version + composer_versions = ['0.18.0'] # 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): diff --git a/setup.py b/setup.py index 2ede354b8d..c58cae7c84 100644 --- a/setup.py +++ b/setup.py @@ -78,8 +78,8 @@ def package_files(prefix: str, directory: str, extension: str): 'tqdm>=4.62.3,<5', 'torchmetrics>=0.10.0,<1.1', 'torch_optimizer>=0.3.0,<0.4', - 'torchvision>=0.13.1,<0.20', # TODO: Tighten before release - 'torch>=1.13.1,<2.3.1', # TODO: Tighten before release + 'torchvision>=0.13.1,<0.17', # TODO: Tighten before release + 'torch>=1.13.1,<2.2.0', # TODO: Tighten before release 'requests>=2.26.0,<3', 'numpy>=1.21.5,<1.27.0', 'psutil>=5.8.0,<6',