diff --git a/.gitlab/stages/01.build.yml b/.gitlab/stages/01.build.yml index f5a80c4074d..aef3d64b014 100644 --- a/.gitlab/stages/01.build.yml +++ b/.gitlab/stages/01.build.yml @@ -64,12 +64,12 @@ test:pre_build_image: - IMAGE: CI_MCORE_DEV_IMAGE FILE: Dockerfile.ci.dev IMAGE_TYPE: dev - BASE_IMAGE: nvcr.io/nvidia/pytorch:26.02-py3 + BASE_IMAGE: nvcr.io/nvidia/pytorch:26.04-py3 PLATFORM: amd64 - IMAGE: CI_MCORE_DEV_IMAGE FILE: Dockerfile.ci.dev IMAGE_TYPE: dev - BASE_IMAGE: nvcr.io/nvidia/pytorch:26.02-py3 + BASE_IMAGE: nvcr.io/nvidia/pytorch:26.04-py3 PLATFORM: arm64 - IMAGE: UTILITY_IMAGE FILE: Dockerfile.linting