From 88aeec3960efd155480947e934e4bc5a9d76659f Mon Sep 17 00:00:00 2001 From: v-chen_data Date: Tue, 19 Nov 2024 10:35:59 -0500 Subject: [PATCH] typo --- docker/README.md | 18 +++++++++--------- docker/build_matrix.yaml | 8 ++++---- 2 files changed, 13 insertions(+), 13 deletions(-) diff --git a/docker/README.md b/docker/README.md index 65fdf796f2..41bd0e51b6 100644 --- a/docker/README.md +++ b/docker/README.md @@ -30,15 +30,15 @@ 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 | 2.5.1 | 12.4.1 (Infiniband) | 3.11 | `mosaicml/pytorch:latest`, `mosaicml/pytorch:2.5.1_cu124-python3.11-ubuntu22.04` | -| Ubuntu 20.04 | Base | 2.5.1 | 12.4.1 (EFA) | 3.11 | `mosaicml/pytorch:latest-aws`, `mosaicml/pytorch:2.5.1_cu124-python3.11-ubuntu22.04-aws` | -| Ubuntu 20.04 | Base | 2.5.1 | cpu | 3.11 | `mosaicml/pytorch:latest_cpu`, `mosaicml/pytorch:2.5.1_cpu-python3.11-ubuntu22.04` | -| Ubuntu 20.04 | Base | 2.4.1 | 12.4.1 (Infiniband) | 3.11 | `mosaicml/pytorch:2.4.1_cu124-python3.11-ubuntu22.04` | -| Ubuntu 20.04 | Base | 2.4.1 | 12.4.1 (EFA) | 3.11 | `mosaicml/pytorch:2.4.1_cu124-python3.11-ubuntu22.04-aws` | -| Ubuntu 20.04 | Base | 2.4.1 | cpu | 3.11 | `mosaicml/pytorch:2.4.1_cpu-python3.11-ubuntu22.04` | -| Ubuntu 20.04 | Base | 2.3.1 | 12.1.1 (Infiniband) | 3.11 | `mosaicml/pytorch:2.3.1_cu121-python3.11-ubuntu22.04` | -| Ubuntu 20.04 | Base | 2.3.1 | 12.1.1 (EFA) | 3.11 | `mosaicml/pytorch:2.3.1_cu121-python3.11-ubuntu22.04-aws` | -| Ubuntu 20.04 | Base | 2.3.1 | cpu | 3.11 | `mosaicml/pytorch:2.3.1_cpu-python3.11-ubuntu22.04` | +| Ubuntu 22.04 | Base | 2.5.1 | 12.4.1 (Infiniband) | 3.11 | `mosaicml/pytorch:latest`, `mosaicml/pytorch:2.5.1_cu124-python3.11-ubuntu22.04` | +| Ubuntu 22.04 | Base | 2.5.1 | 12.4.1 (EFA) | 3.11 | `mosaicml/pytorch:latest-aws`, `mosaicml/pytorch:2.5.1_cu124-python3.11-ubuntu22.04-aws` | +| Ubuntu 22.04 | Base | 2.5.1 | cpu | 3.11 | `mosaicml/pytorch:latest_cpu`, `mosaicml/pytorch:2.5.1_cpu-python3.11-ubuntu22.04` | +| Ubuntu 22.04 | Base | 2.4.1 | 12.4.1 (Infiniband) | 3.11 | `mosaicml/pytorch:2.4.1_cu124-python3.11-ubuntu22.04` | +| Ubuntu 22.04 | Base | 2.4.1 | 12.4.1 (EFA) | 3.11 | `mosaicml/pytorch:2.4.1_cu124-python3.11-ubuntu22.04-aws` | +| Ubuntu 22.04 | Base | 2.4.1 | cpu | 3.11 | `mosaicml/pytorch:2.4.1_cpu-python3.11-ubuntu22.04` | +| Ubuntu 22.04 | Base | 2.3.1 | 12.1.1 (Infiniband) | 3.11 | `mosaicml/pytorch:2.3.1_cu121-python3.11-ubuntu22.04` | +| Ubuntu 22.04 | Base | 2.3.1 | 12.1.1 (EFA) | 3.11 | `mosaicml/pytorch:2.3.1_cu121-python3.11-ubuntu22.04-aws` | +| Ubuntu 22.04 | Base | 2.3.1 | cpu | 3.11 | `mosaicml/pytorch:2.3.1_cpu-python3.11-ubuntu22.04` | **Note**: The `mosaicml/pytorch:latest`, `mosaicml/pytorch:latest_cpu`, and `mosaicml/pytorch:latest-aws` diff --git a/docker/build_matrix.yaml b/docker/build_matrix.yaml index cf24bd0f6a..b062299eca 100644 --- a/docker/build_matrix.yaml +++ b/docker/build_matrix.yaml @@ -34,7 +34,7 @@ TARGET: pytorch_stage TORCHVISION_VERSION: 0.20.1 - AWS_OFI_NCCL_VERSION: '' - BASE_IMAGE: ubuntu:20.04 + BASE_IMAGE: ubuntu:22.04 CUDA_VERSION: '' IMAGE_NAME: torch-2-5-1-cpu MOFED_VERSION: '' @@ -81,7 +81,7 @@ TARGET: pytorch_stage TORCHVISION_VERSION: 0.19.1 - AWS_OFI_NCCL_VERSION: '' - BASE_IMAGE: ubuntu:20.04 + BASE_IMAGE: ubuntu:22.04 CUDA_VERSION: '' IMAGE_NAME: torch-2-4-1-cpu MOFED_VERSION: '' @@ -152,7 +152,7 @@ TARGET: pytorch_stage TORCHVISION_VERSION: 0.18.1 - AWS_OFI_NCCL_VERSION: '' - BASE_IMAGE: ubuntu:20.04 + BASE_IMAGE: ubuntu:22.04 CUDA_VERSION: '' IMAGE_NAME: torch-2-3-1-cpu MOFED_VERSION: '' @@ -185,7 +185,7 @@ TARGET: composer_stage TORCHVISION_VERSION: 0.20.1 - AWS_OFI_NCCL_VERSION: '' - BASE_IMAGE: ubuntu:20.04 + BASE_IMAGE: ubuntu:22.04 COMPOSER_INSTALL_COMMAND: mosaicml[all]==0.27.0 CUDA_VERSION: '' IMAGE_NAME: composer-0-27-0-cpu