update premerge gpu

Signed-off-by: Wenqi Li <wenqil@nvidia.com>
Project-MONAI · Nov 17, 2021 · 1ad89be · 1ad89be
1 parent 86f5684
commit 1ad89be
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/.github/workflows/pythonapp-gpu.yml b/.github/workflows/pythonapp-gpu.yml
@@ -19,17 +19,13 @@ jobs:
     strategy:
       matrix:
         environment:
-          - "PT16+CUDA110"
           - "PT17+CUDA102"
           - "PT17+CUDA110"
           - "PT18+CUDA102"
+          - "PT18+CUDA110"
           - "PT19+CUDA114"
           - "PT110+CUDA102"
         include:
-          - environment: PT16+CUDA110
-            # we explicitly set pytorch to -h to avoid pip install error
-            pytorch: "-h"
-            base: "nvcr.io/nvidia/pytorch:20.07-py3"
           - environment: PT17+CUDA102
             pytorch: "torch==1.7.1 torchvision==0.8.2"
             base: "nvcr.io/nvidia/cuda:10.2-devel-ubuntu18.04"
@@ -40,6 +36,10 @@ jobs:
           - environment: PT18+CUDA102
             pytorch: "torch==1.8.1 torchvision==0.9.1"
             base: "nvcr.io/nvidia/cuda:10.2-devel-ubuntu18.04"
+          - environment: PT18+CUDA110
+            # we explicitly set pytorch to -h to avoid pip install error
+            pytorch: "-h"
+            base: "nvcr.io/nvidia/pytorch:21.02-py3"
           - environment: PT19+CUDA114
             # we explicitly set pytorch to -h to avoid pip install error
             # https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes