From abfb67a79cd46e363bacef70ee30e03d01330e73 Mon Sep 17 00:00:00 2001 From: Nic Ma Date: Tue, 1 Mar 2022 00:30:41 +0800 Subject: [PATCH] [DLMED] update to 22.02 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 b34f2c87ac..b6ed2274ee 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:22.01"] # 21.02 for backward comp. + container: ["pytorch:21.02", "pytorch:22.02"] # 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:22.01"] # 21.02 for backward comp. + container: ["pytorch:21.02", "pytorch:22.02"] # 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 6423befc1c..68fdfde445 100644 --- a/.github/workflows/pythonapp-gpu.yml +++ b/.github/workflows/pythonapp-gpu.yml @@ -45,9 +45,9 @@ jobs: base: "nvcr.io/nvidia/pytorch:21.10-py3" - environment: PT110+CUDA116 # we explicitly set pytorch to -h to avoid pip install error - # 22.01: 1.11.0a0+bfe5ad2 + # 22.02: 1.11.0a0+17540c5 pytorch: "-h" - base: "nvcr.io/nvidia/pytorch:22.01-py3" + base: "nvcr.io/nvidia/pytorch:22.02-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 ab79eaaa5e..4171309c70 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:22.01-py3 +ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:22.02-py3 FROM ${PYTORCH_IMAGE} LABEL maintainer="monai.contact@gmail.com"