nightly changes

mosaicml · Nov 22, 2023 · 11be9f7 · 11be9f7
1 parent a193115
commit 11be9f7
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 5 deletions.
diff --git a/docker/README.md b/docker/README.md
@@ -31,6 +31,7 @@ To install composer, once inside the image, run `pip install mosaicml`.
 <!-- BEGIN_PYTORCH_BUILD_MATRIX -->
 | Linux Distro   | Flavor   | PyTorch Version   | CUDA Version        | Python Version   | Docker Tags                                                                              |
 |----------------|----------|-------------------|---------------------|------------------|------------------------------------------------------------------------------------------|
+| Ubuntu 20.04   | Base     | 2.2.0             | 12.1.0 (Infiniband) | 3.10             | `mosaicml/pytorch:2.2.0_cu121-nightly20231121-python3.10-ubuntu20.04`                    |
 | Ubuntu 20.04   | Base     | 2.2.0             | 12.1.0 (Infiniband) | 3.10             | `mosaicml/pytorch:2.2.0_cu121-nightly20231024-python3.10-ubuntu20.04`                    |
 | Ubuntu 20.04   | Base     | 2.1.0             | 12.1.0 (Infiniband) | 3.10             | `mosaicml/pytorch:latest`, `mosaicml/pytorch:2.1.0_cu121-python3.10-ubuntu20.04`         |
 | Ubuntu 20.04   | Base     | 2.1.0             | 12.1.0 (EFA)        | 3.10             | `mosaicml/pytorch:latest-aws`, `mosaicml/pytorch:2.1.0_cu121-python3.10-ubuntu20.04-aws` |

diff --git a/docker/build_matrix.yaml b/docker/build_matrix.yaml
@@ -151,6 +151,19 @@
   - mosaicml/pytorch:2.2.0_cu121-nightly20231024-python3.10-ubuntu20.04
   TARGET: pytorch_stage
   TORCHVISION_VERSION: 0.17.0
+  - AWS_OFI_NCCL_VERSION: ''
+  BASE_IMAGE: nvidia/cuda:12.1.0-cudnn8-devel-ubuntu20.04
+  CUDA_VERSION: 12.1.0
+  IMAGE_NAME: torch-nightly-2-2-0-20231121-cu121
+  MOFED_VERSION: 5.5-1.0.3.2
+  PYTHON_VERSION: '3.10'
+  PYTORCH_NIGHTLY_URL: https://download.pytorch.org/whl/nightly/cu121
+  PYTORCH_NIGHTLY_VERSION: dev20231121+cu121
+  PYTORCH_VERSION: 2.2.0
+  TAGS:
+  - mosaicml/pytorch:2.2.0_cu121-nightly20231121-python3.10-ubuntu20.04
+  TARGET: pytorch_stage
+  TORCHVISION_VERSION: 0.17.0
 - AWS_OFI_NCCL_VERSION: ''
   BASE_IMAGE: nvidia/cuda:12.1.0-cudnn8-devel-ubuntu20.04
   COMPOSER_INSTALL_COMMAND: mosaicml[all]==0.17.0

diff --git a/docker/generate_build_matrix.py b/docker/generate_build_matrix.py
@@ -200,15 +200,15 @@ def _main():
         'AWS_OFI_NCCL_VERSION': '',
         'BASE_IMAGE': 'nvidia/cuda:12.1.0-cudnn8-devel-ubuntu20.04',
         'CUDA_VERSION': '12.1.0',
-        'IMAGE_NAME': 'torch-nightly-2-2-0-20231024-cu121',
+        'IMAGE_NAME': 'torch-nightly-2-2-0-20231121-cu121',
         'MOFED_VERSION': '5.5-1.0.3.2',
         'PYTHON_VERSION': '3.10',
-        'PYTORCH_VERSION': '2.2.0',
         'PYTORCH_NIGHTLY_URL': 'https://download.pytorch.org/whl/nightly/cu121',
-        'PYTORCH_NIGHTLY_VERSION': 'dev20231024+cu121',
-        'TAGS': ['mosaicml/pytorch:2.2.0_cu121-nightly20231024-python3.10-ubuntu20.04'],
+        'PYTORCH_NIGHTLY_VERSION': 'dev20231121+cu121',
+        'PYTORCH_VERSION': '2.2.0',
+        'TAGS': 'mosaicml/pytorch:2.2.0_cu121-nightly20231121-python3.10-ubuntu20.04',
         'TARGET': 'pytorch_stage',
-        'TORCHVISION_VERSION': '0.17.0'
+        'TORCHVISION_VERSION': '0.17.0',
     }
     pytorch_entries.append(nightly_entry)
     composer_entries = []