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 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: 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 }} 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 diff --git a/docker/README.md b/docker/README.md index 246c6f8baa..b36ffe6492 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 \ @@ -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 \ . ```