fixes #3660 2003->2106

Signed-off-by: Wenqi Li <wenqil@nvidia.com>
Project-MONAI · Jan 13, 2022 · a8fbc70 · a8fbc70
1 parent 93f6f4d
commit a8fbc70
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/cron.yml b/.github/workflows/cron.yml
@@ -10,7 +10,7 @@ jobs:
   cron-gpu:
     if: github.repository == 'Project-MONAI/MONAI'
     container:
-      image: nvcr.io/nvidia/pytorch:20.03-py3  # CUDA 10.2
+      image: nvcr.io/nvidia/pytorch:21.06-py3  # CUDA 11.3
       options: "--gpus all"
     runs-on: [self-hosted, linux, x64, common]
     strategy:

diff --git a/.github/workflows/setupapp.yml b/.github/workflows/setupapp.yml
@@ -21,7 +21,7 @@ jobs:
   coverage-py3:
     if: github.repository == 'Project-MONAI/MONAI'
     container:
-      image: nvcr.io/nvidia/pytorch:20.03-py3  # CUDA 10.2
+      image: nvcr.io/nvidia/pytorch:21.06-py3  # CUDA 11.3
       options: --gpus all
     runs-on: [self-hosted, linux, x64, common]
     steps: