diff --git a/.github/workflows/cron.yml b/.github/workflows/cron.yml index 761b1f7ebc..3562672232 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.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: diff --git a/.github/workflows/pythonapp.yml b/.github/workflows/pythonapp.yml index e5803028a0..738d657211 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.02-py3" + base: "nvcr.io/nvidia/pytorch:21.03-py3" container: image: ${{ matrix.base }} options: --gpus all diff --git a/Dockerfile b/Dockerfile index 57ea567869..54d1f02275 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.02-py3 +ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:21.03-py3 FROM ${PYTORCH_IMAGE} LABEL maintainer="monai.contact@gmail.com"