From ad4bccc951f40a9c0f9739671ffe4f09fcdd82f5 Mon Sep 17 00:00:00 2001 From: Nic Ma Date: Wed, 22 Dec 2021 11:27:27 +0800 Subject: [PATCH 1/3] [DLMED] update to 21.12 Signed-off-by: Nic Ma --- .github/workflows/cron.yml | 4 ++-- .github/workflows/pythonapp-gpu.yml | 4 ++-- Dockerfile | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/.github/workflows/cron.yml b/.github/workflows/cron.yml index 835069c833..70585eade0 100644 --- 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:21.10"] # 21.02 for backward comp. + container: ["pytorch:21.02", "pytorch:21.12"] # 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:21.10"] # 21.02 for backward comp. + container: ["pytorch:21.02", "pytorch:21.12"] # 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/pythonapp-gpu.yml b/.github/workflows/pythonapp-gpu.yml index c5cd70af7f..5bf11d281b 100644 --- a/.github/workflows/pythonapp-gpu.yml +++ b/.github/workflows/pythonapp-gpu.yml @@ -43,9 +43,9 @@ jobs: - environment: PT19+CUDA114 # we explicitly set pytorch to -h to avoid pip install error # https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes - # 21.10: 1.10.0a0+0aef44c + # 21.12: 1.11.0a0+b6df043 pytorch: "-h" - base: "nvcr.io/nvidia/pytorch:21.10-py3" + base: "nvcr.io/nvidia/pytorch:21.12-py3" - environment: PT110+CUDA102 pytorch: "torch==1.10.1 torchvision==0.11.2" base: "nvcr.io/nvidia/cuda:10.2-devel-ubuntu18.04" diff --git a/Dockerfile b/Dockerfile index 2d889c8b45..43dc103e0f 100644 --- 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:21.10-py3 +ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:21.12-py3 FROM ${PYTORCH_IMAGE} LABEL maintainer="monai.contact@gmail.com" From d7f00f00e9e7bb026a8e4d993dd1d4167e406de0 Mon Sep 17 00:00:00 2001 From: Nic Ma Date: Wed, 22 Dec 2021 14:33:49 +0800 Subject: [PATCH 2/3] [DLMED] update according to comments Signed-off-by: Nic Ma --- .github/workflows/pythonapp-gpu.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/pythonapp-gpu.yml b/.github/workflows/pythonapp-gpu.yml index 5bf11d281b..1bf71783fa 100644 --- a/.github/workflows/pythonapp-gpu.yml +++ b/.github/workflows/pythonapp-gpu.yml @@ -23,7 +23,7 @@ jobs: - "PT17+CUDA110" - "PT18+CUDA102" - "PT18+CUDA110" - - "PT19+CUDA114" + - "PT110+CUDA115" - "PT110+CUDA102" include: - environment: PT17+CUDA102 @@ -40,7 +40,7 @@ jobs: # we explicitly set pytorch to -h to avoid pip install error pytorch: "-h" base: "nvcr.io/nvidia/pytorch:21.02-py3" - - environment: PT19+CUDA114 + - environment: PT110+CUDA115 # we explicitly set pytorch to -h to avoid pip install error # https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes # 21.12: 1.11.0a0+b6df043 From 4cc7c1a1667065be243bf49c90f6730ebda254a6 Mon Sep 17 00:00:00 2001 From: Nic Ma Date: Wed, 22 Dec 2021 15:07:54 +0800 Subject: [PATCH 3/3] [DLMED] add PyTorch 1.9 test Signed-off-by: Nic Ma --- .github/workflows/pythonapp-gpu.yml | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/.github/workflows/pythonapp-gpu.yml b/.github/workflows/pythonapp-gpu.yml index 1bf71783fa..bae009d0a1 100644 --- a/.github/workflows/pythonapp-gpu.yml +++ b/.github/workflows/pythonapp-gpu.yml @@ -22,7 +22,7 @@ jobs: - "PT17+CUDA102" - "PT17+CUDA110" - "PT18+CUDA102" - - "PT18+CUDA110" + - "PT19+CUDA114" - "PT110+CUDA115" - "PT110+CUDA102" include: @@ -36,10 +36,11 @@ 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 + - environment: PT19+CUDA114 # we explicitly set pytorch to -h to avoid pip install error + # 21.10: 1.10.0a0+0aef44c pytorch: "-h" - base: "nvcr.io/nvidia/pytorch:21.02-py3" + base: "nvcr.io/nvidia/pytorch:21.10-py3" - environment: PT110+CUDA115 # we explicitly set pytorch to -h to avoid pip install error # https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes