From 41873b39b88cab3b3d4ab269aa64698c8fadf121 Mon Sep 17 00:00:00 2001 From: Wenqi Li Date: Fri, 2 Apr 2021 17:41:13 +0100 Subject: [PATCH] Revert "1919 - test pt 2103 (#1920)" This reverts commit 4bd26f9f6c18e7c6e68320d19f556705d9afab60. Signed-off-by: Wenqi Li --- .github/workflows/cron.yml | 4 ++-- .github/workflows/pythonapp.yml | 2 +- Dockerfile | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/cron.yml b/.github/workflows/cron.yml index 3562672232..761b1f7ebc 100644 --- a/.github/workflows/cron.yml +++ b/.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.03-py3 # testing with the latest pytorch base image + image: nvcr.io/nvidia/pytorch:21.02-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.03-py3 # testing with the latest pytorch base image + image: nvcr.io/nvidia/pytorch:21.02-py3 # testing with the latest pytorch base image options: "--gpus all --ipc=host" runs-on: [self-hosted, linux, x64, common] steps: diff --git a/.github/workflows/pythonapp.yml b/.github/workflows/pythonapp.yml index 738d657211..e5803028a0 100644 --- a/.github/workflows/pythonapp.yml +++ b/.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.03-py3" + base: "nvcr.io/nvidia/pytorch:21.02-py3" container: image: ${{ matrix.base }} options: --gpus all diff --git a/Dockerfile b/Dockerfile index 54d1f02275..57ea567869 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.03-py3 +ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:21.02-py3 FROM ${PYTORCH_IMAGE} LABEL maintainer="monai.contact@gmail.com"