1919 - test pt 2103 (Project-MONAI#1920)

* update to use pytorch2103

Signed-off-by: Wenqi Li <[email protected]>

Author: wyli

.github/workflows/cron.yml

@@ -60,7 +60,7 @@ jobs:
   cron-pt-image:
     if: github.repository == 'Project-MONAI/MONAI'
     container:
-      image: nvcr.io/nvidia/pytorch:21.02-py3  # testing with the latest pytorch base image
+      image: nvcr.io/nvidia/pytorch:21.03-py3  # testing with the latest pytorch base image
       options: "--gpus all"
     runs-on: [self-hosted, linux, x64, common]
     steps:
@@ -133,7 +133,7 @@ jobs:
     if: github.repository == 'Project-MONAI/MONAI'
     needs: cron-gpu  # so that monai itself is verified first
     container:
-      image: nvcr.io/nvidia/pytorch:21.02-py3  # testing with the latest pytorch base image
+      image: nvcr.io/nvidia/pytorch:21.03-py3  # testing with the latest pytorch base image
       options: "--gpus all --ipc=host"
     runs-on: [self-hosted, linux, x64, common]
     steps:

.github/workflows/pythonapp.yml

@@ -186,7 +186,7 @@ jobs:
           - environment: PT18+CUDA112
             # we explicitly set pytorch to -h to avoid pip install error
             pytorch: "-h"
-            base: "nvcr.io/nvidia/pytorch:21.02-py3"
+            base: "nvcr.io/nvidia/pytorch:21.03-py3"
     container:
       image: ${{ matrix.base }}
       options: --gpus all

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:21.02-py3
+ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:21.03-py3
 FROM ${PYTORCH_IMAGE}
 
 LABEL maintainer="[email protected]"