From 39e7e603aa0cd89ebe8ebde80be6b3d866301601 Mon Sep 17 00:00:00 2001 From: YanxuanLiu Date: Thu, 30 Mar 2023 14:12:37 +0800 Subject: [PATCH 1/2] upgrade pytorch to 23.03 Signed-off-by: YanxuanLiu --- .github/workflows/cron.yml | 8 ++++---- .github/workflows/pythonapp-gpu.yml | 4 ++-- Dockerfile | 2 +- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/.github/workflows/cron.yml b/.github/workflows/cron.yml index 88f4522b5b..ce9e8a4394 100644 --- a/.github/workflows/cron.yml +++ b/.github/workflows/cron.yml @@ -30,7 +30,7 @@ jobs: base: "nvcr.io/nvidia/pytorch:21.06-py3" # CUDA 11.3 - environment: PTLATEST+CUDA118 pytorch: "-U torch torchvision --extra-index-url https://download.pytorch.org/whl/cu118" - base: "nvcr.io/nvidia/pytorch:23.02-py3" # CUDA 11.8 + base: "nvcr.io/nvidia/pytorch:23.03-py3" # CUDA 11.8 container: image: ${{ matrix.base }} options: "--gpus all" @@ -76,7 +76,7 @@ jobs: if: github.repository == 'Project-MONAI/MONAI' strategy: matrix: - container: ["pytorch:22.09", "pytorch:22.11", "pytorch:23.02"] + container: ["pytorch:22.09", "pytorch:22.11", "pytorch:23.03"] container: image: nvcr.io/nvidia/${{ matrix.container }}-py3 # testing with the latest pytorch base image options: "--gpus all" @@ -121,7 +121,7 @@ jobs: if: github.repository == 'Project-MONAI/MONAI' strategy: matrix: - container: ["pytorch:22.09", "pytorch:22.11", "pytorch:23.02"] + container: ["pytorch:21.02", "pytorch:21.10", "pytorch:23.03"] # 21.02, 21.10 for backward comp. container: image: nvcr.io/nvidia/${{ matrix.container }}-py3 # testing with the latest pytorch base image options: "--gpus all" @@ -221,7 +221,7 @@ jobs: if: github.repository == 'Project-MONAI/MONAI' needs: cron-gpu # so that monai itself is verified first container: - image: nvcr.io/nvidia/pytorch:23.02-py3 # testing with the latest pytorch base image + image: nvcr.io/nvidia/pytorch:23.03-py3 # testing with the latest pytorch base image options: "--gpus all --ipc=host" runs-on: [self-hosted, linux, x64, integration] steps: diff --git a/.github/workflows/pythonapp-gpu.yml b/.github/workflows/pythonapp-gpu.yml index acbd6c648d..164df5efcc 100644 --- a/.github/workflows/pythonapp-gpu.yml +++ b/.github/workflows/pythonapp-gpu.yml @@ -42,9 +42,9 @@ jobs: pytorch: "torch==1.13.1 torchvision==0.14.1" base: "nvcr.io/nvidia/cuda:11.6.1-devel-ubuntu18.04" - environment: PT114+CUDA120DOCKER - # 23.02: 1.14.0a0+44dac51 + # 23.03: 2.0.0a0+1767026 pytorch: "-h" # we explicitly set pytorch to -h to avoid pip install error - base: "nvcr.io/nvidia/pytorch:23.02-py3" + base: "nvcr.io/nvidia/pytorch:23.03-py3" container: image: ${{ matrix.base }} options: --gpus all --env NVIDIA_DISABLE_REQUIRE=true # workaround for unsatisfied condition: cuda>=11.6 diff --git a/Dockerfile b/Dockerfile index 9bc0eee70c..653dd1571c 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:23.02-py3 +ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:23.03-py3 FROM ${PYTORCH_IMAGE} LABEL maintainer="monai.contact@gmail.com" From 290c512d88dd89ac78b261b156c4e0a5e312b496 Mon Sep 17 00:00:00 2001 From: YanxuanLiu Date: Thu, 30 Mar 2023 14:18:23 +0800 Subject: [PATCH 2/2] fix Signed-off-by: YanxuanLiu --- .github/workflows/cron.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/cron.yml b/.github/workflows/cron.yml index ce9e8a4394..c1015cd541 100644 --- a/.github/workflows/cron.yml +++ b/.github/workflows/cron.yml @@ -121,7 +121,7 @@ jobs: if: github.repository == 'Project-MONAI/MONAI' strategy: matrix: - container: ["pytorch:21.02", "pytorch:21.10", "pytorch:23.03"] # 21.02, 21.10 for backward comp. + container: ["pytorch:22.09", "pytorch:22.11", "pytorch:23.03"] container: image: nvcr.io/nvidia/${{ matrix.container }}-py3 # testing with the latest pytorch base image options: "--gpus all"