From fe806d386c4f4937302900807033c66b425e3507 Mon Sep 17 00:00:00 2001
From: "nemo-autobot-origin[bot]"
 <297562586+nemo-autobot-origin[bot]@users.noreply.github.com>
Date: Wed, 1 Jul 2026 19:21:21 +0000
Subject: [PATCH 1/6] ci: bump Dockerfile.ci base image to pytorch 26.06

---
 docker/Dockerfile.ci | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/Dockerfile.ci b/docker/Dockerfile.ci
index e0be6c1ec4..beb057612c 100644
--- a/docker/Dockerfile.ci
+++ b/docker/Dockerfile.ci
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:26.04-py3
+ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:26.06-py3
 FROM ${BASE_IMAGE} AS megatron_bridge
 WORKDIR /opt/Megatron-Bridge
 

From ce787a30f2696a8dc1b53f61c191107ea0b95ce4 Mon Sep 17 00:00:00 2001
From: "nemo-autobot-origin[bot]"
 <297562586+nemo-autobot-origin[bot]@users.noreply.github.com>
Date: Wed, 1 Jul 2026 20:32:33 +0000
Subject: [PATCH 2/6] ci: bump pytorch custom-container to 26.06 in release
 workflow

---
 .github/workflows/release.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/release.yaml b/.github/workflows/release.yaml
index 0b64eb9893..e2d098d1e8 100644
--- a/.github/workflows/release.yaml
+++ b/.github/workflows/release.yaml
@@ -99,7 +99,7 @@ jobs:
       packaging: uv
       has-src-dir: true
       skip-test-wheel: true
-      custom-container: nvcr.io/nvidia/pytorch:25.11-py3
+      custom-container: nvcr.io/nvidia/pytorch:26.06-py3
       runner: ${{ needs.pre-flight.outputs.runner_prefix }}-container
       no-build-isolation: true
       app-id: ${{ vars.BOT_ID }}

From 7545f3dfd0947dda398d73a50f2be71397f0a85e Mon Sep 17 00:00:00 2001
From: "nemo-autobot-origin[bot]"
 <297562586+nemo-autobot-origin[bot]@users.noreply.github.com>
Date: Wed, 1 Jul 2026 20:32:40 +0000
Subject: [PATCH 3/6] ci: bump pytorch base image to 26.06 in cicd-main
 workflow

---
 .github/workflows/cicd-main.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/cicd-main.yml b/.github/workflows/cicd-main.yml
index 51bc5fa374..0fa5ce8856 100644
--- a/.github/workflows/cicd-main.yml
+++ b/.github/workflows/cicd-main.yml
@@ -438,7 +438,7 @@ jobs:
           context: .
           platforms: ${{ steps.platform.outputs.platforms }}
           build-args: |
-            FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:25.09-py3
+            FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:26.06-py3
             MCORE_TRIGGERED_TESTING=${{ env.MCORE_TRIGGERED_TESTING || 'false' }}
             MCORE_COMMIT_SHA=${{ env.MCORE_COMMIT_SHA || 'unknown' }}
           cache-from: type=registry,ref=${{ matrix.registry }}/megatron-bridge:${{ steps.cache_keys.outputs.seed }}-buildcache,mode=max

From 39e7724c2f32a143f8c78615b01559c660e16ec0 Mon Sep 17 00:00:00 2001
From: "nemo-autobot-origin[bot]"
 <297562586+nemo-autobot-origin[bot]@users.noreply.github.com>
Date: Wed, 1 Jul 2026 20:32:53 +0000
Subject: [PATCH 4/6] ci: bump pytorch base image to 26.06 in install-test
 workflow

---
 .github/workflows/install-test.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/install-test.yml b/.github/workflows/install-test.yml
index 7378dc0af6..510e7294dd 100644
--- a/.github/workflows/install-test.yml
+++ b/.github/workflows/install-test.yml
@@ -127,7 +127,7 @@ jobs:
     runs-on: ${{ needs.pre-flight.outputs.runner_prefix }}-container
     name: UV - Python${{ matrix.python-version }} - AMD64/Linux - NGC PyTorch
     container:
-      image: nvcr.io/nvidia/pytorch:25.05-py3
+      image: nvcr.io/nvidia/pytorch:26.06-py3
     strategy:
       fail-fast: false
       matrix:

From 8e04a099cc6bc79ec19c0fff59dbc9f028be8b2e Mon Sep 17 00:00:00 2001
From: "nemo-autobot-origin[bot]"
 <297562586+nemo-autobot-origin[bot]@users.noreply.github.com>
Date: Wed, 1 Jul 2026 20:33:07 +0000
Subject: [PATCH 5/6] docs: bump pytorch base image to 26.06 in docker README
 examples

---
 docker/README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/README.md b/docker/README.md
index 246c6f8baa..3a49c4ab16 100644
--- a/docker/README.md
+++ b/docker/README.md
@@ -52,7 +52,7 @@ docker buildx build \
   --target nemo_fw_base_final \
   --build-arg FW_DEP_BUILDER=trtllm_builder \
   --build-arg FW_BASE_FINAL=trtllm_install \
-  --build-arg NEMO_FW_BASE_IMAGE=nvcr.io/nvidia/pytorch:26.02-py3 \
+  --build-arg NEMO_FW_BASE_IMAGE=nvcr.io/nvidia/pytorch:26.06-py3 \
   --build-arg TRT_LLM_COMMIT=v1.3.0rc4 \
   --build-arg VLLM_VERSION=v0.14.1 \
   -t fw-base:latest \

From 7cebf29cf277c1b00b41820636b96fa7560024ff Mon Sep 17 00:00:00 2001
From: "nemo-autobot-origin[bot]"
 <297562586+nemo-autobot-origin[bot]@users.noreply.github.com>
Date: Wed, 1 Jul 2026 20:33:19 +0000
Subject: [PATCH 6/6] docs: bump pytorch base image to 26.06 in docker README
 (no-TRT-LLM example)

---
 docker/README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/README.md b/docker/README.md
index 3a49c4ab16..b36ffe6492 100644
--- a/docker/README.md
+++ b/docker/README.md
@@ -67,7 +67,7 @@ docker buildx build \
   --target nemo_fw_base_final \
   --build-arg FW_DEP_BUILDER=base \
   --build-arg FW_BASE_FINAL=fw_toolkit_builder \
-  --build-arg NEMO_FW_BASE_IMAGE=nvcr.io/nvidia/pytorch:26.02-py3 \
+  --build-arg NEMO_FW_BASE_IMAGE=nvcr.io/nvidia/pytorch:26.06-py3 \
   -t fw-base:latest \
   .
 ```