From e0b3c63f03c44fc9e9aed1397ce43c1fc0ab54a8 Mon Sep 17 00:00:00 2001 From: Chuck Tang Date: Thu, 8 Aug 2024 17:27:14 -0700 Subject: [PATCH] commit change --- docker/README.md | 4 ++-- docker/build_matrix.yaml | 20 ++++++++++---------- 2 files changed, 12 insertions(+), 12 deletions(-) diff --git a/docker/README.md b/docker/README.md index 1b68cdec39..bd4284b4de 100644 --- a/docker/README.md +++ b/docker/README.md @@ -30,8 +30,8 @@ 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.4.0 | 12.1.4 (Infiniband) | 3.11 | `mosaicml/pytorch:latest`, `mosaicml/pytorch:2.4.0_cu121-python3.11-ubuntu20.04` | -| Ubuntu 20.04 | Base | 2.4.0 | 12.1.4 (EFA) | 3.11 | `mosaicml/pytorch:latest-aws`, `mosaicml/pytorch:2.4.0_cu121-python3.11-ubuntu20.04-aws` | +| Ubuntu 20.04 | Base | 2.4.0 | 12.4.0 (Infiniband) | 3.11 | `mosaicml/pytorch:latest`, `mosaicml/pytorch:2.4.0_cu124-python3.11-ubuntu20.04` | +| Ubuntu 20.04 | Base | 2.4.0 | 12.4.0 (EFA) | 3.11 | `mosaicml/pytorch:latest-aws`, `mosaicml/pytorch:2.4.0_cu124-python3.11-ubuntu20.04-aws` | | Ubuntu 20.04 | Base | 2.4.0 | cpu | 3.11 | `mosaicml/pytorch:latest_cpu`, `mosaicml/pytorch:2.4.0_cpu-python3.11-ubuntu20.04` | | Ubuntu 20.04 | Base | 2.3.1 | 12.1.1 (Infiniband) | 3.11 | `mosaicml/pytorch:2.3.1_cu121-python3.11-ubuntu20.04` | | Ubuntu 20.04 | Base | 2.3.1 | 12.1.1 (EFA) | 3.11 | `mosaicml/pytorch:2.3.1_cu121-python3.11-ubuntu20.04-aws` | diff --git a/docker/build_matrix.yaml b/docker/build_matrix.yaml index 6f4f14ad76..f23ae70bb4 100644 --- a/docker/build_matrix.yaml +++ b/docker/build_matrix.yaml @@ -1,8 +1,8 @@ # This file is automatically generated by generate_build_matrix.py. DO NOT EDIT! - AWS_OFI_NCCL_VERSION: '' - BASE_IMAGE: nvidia/cuda:12.1.4-cudnn8-devel-ubuntu20.04 - CUDA_VERSION: 12.1.4 - IMAGE_NAME: torch-2-4-0-cu121 + BASE_IMAGE: nvidia/cuda:12.4.0-cudnn8-devel-ubuntu20.04 + CUDA_VERSION: 12.4.0 + IMAGE_NAME: torch-2-4-0-cu124 MOFED_VERSION: latest-23.10 NVIDIA_REQUIRE_CUDA_OVERRIDE: '' PYTHON_VERSION: '3.11' @@ -10,14 +10,14 @@ PYTORCH_NIGHTLY_VERSION: '' PYTORCH_VERSION: 2.4.0 TAGS: - - mosaicml/pytorch:2.4.0_cu121-python3.11-ubuntu20.04 + - mosaicml/pytorch:2.4.0_cu124-python3.11-ubuntu20.04 - mosaicml/pytorch:latest TARGET: pytorch_stage TORCHVISION_VERSION: 0.19.0 - AWS_OFI_NCCL_VERSION: v1.9.1-aws - BASE_IMAGE: nvidia/cuda:12.1.4-cudnn8-devel-ubuntu20.04 - CUDA_VERSION: 12.1.4 - IMAGE_NAME: torch-2-4-0-cu121-aws + BASE_IMAGE: nvidia/cuda:12.4.0-cudnn8-devel-ubuntu20.04 + CUDA_VERSION: 12.4.0 + IMAGE_NAME: torch-2-4-0-cu124-aws MOFED_VERSION: '' NVIDIA_REQUIRE_CUDA_OVERRIDE: '' PYTHON_VERSION: '3.11' @@ -25,7 +25,7 @@ PYTORCH_NIGHTLY_VERSION: '' PYTORCH_VERSION: 2.4.0 TAGS: - - mosaicml/pytorch:2.4.0_cu121-python3.11-ubuntu20.04-aws + - mosaicml/pytorch:2.4.0_cu124-python3.11-ubuntu20.04-aws - mosaicml/pytorch:latest-aws TARGET: pytorch_stage TORCHVISION_VERSION: 0.19.0 @@ -181,9 +181,9 @@ TARGET: pytorch_stage TORCHVISION_VERSION: 0.17.2 - AWS_OFI_NCCL_VERSION: '' - BASE_IMAGE: nvidia/cuda:12.1.4-cudnn8-devel-ubuntu20.04 + BASE_IMAGE: nvidia/cuda:12.4.0-cudnn8-devel-ubuntu20.04 COMPOSER_INSTALL_COMMAND: mosaicml[all]==0.23.5 - CUDA_VERSION: 12.1.4 + CUDA_VERSION: 12.4.0 IMAGE_NAME: composer-0-23-5 MOFED_VERSION: latest-23.10 NVIDIA_REQUIRE_CUDA_OVERRIDE: ''