Update base image to 2408

.github/workflows/cron.yml

@@ -13,24 +13,24 @@ jobs:
     strategy:
       matrix:
         environment:
-          - "PT191+CUDA113"
           - "PT110+CUDA113"
-          - "PT113+CUDA113"
-          - "PTLATEST+CUDA121"
+          - "PT113+CUDA118"
+          - "PT210+CUDA121"
+          - "PTLATEST+CUDA124"
         include:
           # https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes
           - environment: PT110+CUDA113
             pytorch: "torch==1.10.2 torchvision==0.11.3 --extra-index-url https://download.pytorch.org/whl/cu113"
             base: "nvcr.io/nvidia/pytorch:21.06-py3"  # CUDA 11.3
-          - environment: PT113+CUDA113
-            pytorch: "torch==1.13.1 torchvision==0.14.1 --extra-index-url https://download.pytorch.org/whl/cu113"
-            base: "nvcr.io/nvidia/pytorch:21.06-py3"  # CUDA 11.3
-          - environment: PT113+CUDA122
+          - environment: PT113+CUDA118
             pytorch: "torch==1.13.1 torchvision==0.14.1 --extra-index-url https://download.pytorch.org/whl/cu121"
-            base: "nvcr.io/nvidia/pytorch:23.08-py3"  # CUDA 12.2
+            base: "nvcr.io/nvidia/pytorch:22.10-py3"  # CUDA 11.8
+          - environment: PT210+CUDA121
+            pytorch: "pytorch==2.1.0 torchvision==0.16.0 --extra-index-url https://download.pytorch.org/whl/cu121"
+            base: "nvcr.io/nvidia/pytorch:23.08-py3"  # CUDA 12.1
           - environment: PTLATEST+CUDA124
             pytorch: "-U torch torchvision --extra-index-url https://download.pytorch.org/whl/cu121"
-            base: "nvcr.io/nvidia/pytorch:24.03-py3"  # CUDA 12.4
+            base: "nvcr.io/nvidia/pytorch:24.08-py3"  # CUDA 12.4
     container:
       image: ${{ matrix.base }}
       options: "--gpus all"
@@ -80,7 +80,7 @@ jobs:
     if: github.repository == 'Project-MONAI/MONAI'
     strategy:
       matrix:
-        container: ["pytorch:23.08", "pytorch:24.03"]
+        container: ["pytorch:23.08", "pytorch:24.08"]
     container:
       image: nvcr.io/nvidia/${{ matrix.container }}-py3  # testing with the latest pytorch base image
       options: "--gpus all"
@@ -129,7 +129,7 @@ jobs:
     if: github.repository == 'Project-MONAI/MONAI'
     strategy:
       matrix:
-        container: ["pytorch:24.03"]
+        container: ["pytorch:24.08"]
     container:
       image: nvcr.io/nvidia/${{ matrix.container }}-py3  # testing with the latest pytorch base image
       options: "--gpus all"
@@ -233,7 +233,7 @@ jobs:
     if: github.repository == 'Project-MONAI/MONAI'
     needs: cron-gpu  # so that monai itself is verified first
     container:
-      image: nvcr.io/nvidia/pytorch:24.03-py3  # testing with the latest pytorch base image
+      image: nvcr.io/nvidia/pytorch:24.08-py3  # testing with the latest pytorch base image
       options: "--gpus all --ipc=host"
     runs-on: [self-hosted, linux, x64, integration]
     steps:

.github/workflows/pythonapp-gpu.yml

@@ -44,9 +44,9 @@ jobs:
             pytorch: "-h"  # we explicitly set pytorch to -h to avoid pip install error
             base: "nvcr.io/nvidia/pytorch:23.08-py3"
           - environment: PT210+CUDA121DOCKER
-            # 24.03: 2.3.0a0+40ec155e58.nv24.3
+            # 24.08: 2.3.0a0+40ec155e58.nv24.3
             pytorch: "-h"  # we explicitly set pytorch to -h to avoid pip install error
-            base: "nvcr.io/nvidia/pytorch:24.03-py3"
+            base: "nvcr.io/nvidia/pytorch:24.08-py3"
     container:
       image: ${{ matrix.base }}
       options: --gpus all --env NVIDIA_DISABLE_REQUIRE=true  # workaround for unsatisfied condition: cuda>=11.6

Dockerfile

@@ -11,7 +11,7 @@
 
 # To build with a different base image
 # please run `docker build` using the `--build-arg PYTORCH_IMAGE=...` flag.
-ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:24.03-py3
+ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:24.08-py3
 FROM ${PYTORCH_IMAGE}
 
 LABEL maintainer="monai.contact@gmail.com"