Workaround OOM issue in classification 2D integration tests (#3949)

Nic-Ma · web-flow · commit 6ea97423b969 · 2022-03-16T20:57:59.000Z
* [DLMED] update to 21.10

Signed-off-by: Nic Ma &lt;nma@nvidia.com&gt;

* [DLMED] fix typo in integration test config

Signed-off-by: Nic Ma &lt;nma@nvidia.com&gt;
diff --git a/.github/workflows/cron.yml b/.github/workflows/cron.yml
@@ -62,7 +62,7 @@ jobs:
     if: github.repository == 'Project-MONAI/MONAI'
     strategy:
       matrix:
-        container: ["pytorch:21.02", "pytorch:22.02"]  # 21.02 for backward comp.
+        container: ["pytorch:21.02", "pytorch:21.10"]  # 21.02 for backward comp.
     container:
       image: nvcr.io/nvidia/${{ matrix.container }}-py3  # testing with the latest pytorch base image
       options: "--gpus all"
@@ -106,7 +106,7 @@ jobs:
     if: github.repository == 'Project-MONAI/MONAI'
     strategy:
       matrix:
-        container: ["pytorch:21.02", "pytorch:22.02"]  # 21.02 for backward comp.
+        container: ["pytorch:21.02", "pytorch:21.10"]  # 21.02 for backward comp.
     container:
       image: nvcr.io/nvidia/${{ matrix.container }}-py3  # testing with the latest pytorch base image
       options: "--gpus all"
diff --git a/.github/workflows/integration.yml b/.github/workflows/integration.yml
@@ -7,7 +7,7 @@ on:
 jobs:
   integration-py3:
     container:
-      image: nvcr.io/nvidia/pytorch:21.12-py3  # CUDA 11.5
+      image: nvcr.io/nvidia/pytorch:21.10-py3  # CUDA 11.4
       options: --gpus all
     runs-on: [self-hosted, linux, x64, common]
     steps:
@@ -34,7 +34,7 @@ jobs:
         which python
         python -m pip install --upgrade pip wheel
         python -m pip uninstall -y torch torchvision
-        python -m pip install torch==1.11.0+cu115 torchvision==0.12.0+cu115 -f https://download.pytorch.org/whl/torch_stable.html
+        python -m pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html
         python -m pip install -r requirements-dev.txt
     - name: Run integration tests
       run: |
diff --git a/Dockerfile b/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:22.02-py3
+ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:21.10-py3
 FROM ${PYTORCH_IMAGE}
 
 LABEL maintainer="monai.contact@gmail.com"
