Trying older pytorch version

Author: edwardsp

.github/workflows/build_nccl_image_dev.yaml

@@ -47,9 +47,9 @@ jobs:
       with:
         images: ghcr.io/azure/ai-infrastructure-on-azure/nccl-test
         tags: |
-          type=raw,value=dev
-          type=raw,value=dev-{{sha}}
-          type=raw,value=dev-pytorch-{{date 'YYYYMMDD-hhmmss' tz='UTC'}}
+          type=raw,value=dev2503
+          type=raw,value=dev2503-{{sha}}
+          type=raw,value=dev2503-pytorch-{{date 'YYYYMMDD-hhmmss' tz='UTC'}}
 
     - name: Build and push Dev Image
       uses: docker/build-push-action@v6
@@ -67,14 +67,14 @@ jobs:
         echo "## Dev NCCL Test Image Built Successfully! 🚀" >> $GITHUB_STEP_SUMMARY
         echo "" >> $GITHUB_STEP_SUMMARY
         echo "**Image Tags:**" >> $GITHUB_STEP_SUMMARY
-        echo "- \`ghcr.io/azure/ai-infrastructure-on-azure/nccl-test:dev\`" >> $GITHUB_STEP_SUMMARY
-        echo "- \`ghcr.io/azure/ai-infrastructure-on-azure/nccl-test:dev-${{ github.sha }}\`" >> $GITHUB_STEP_SUMMARY
+        echo "- \`ghcr.io/azure/ai-infrastructure-on-azure/nccl-test:dev2503\`" >> $GITHUB_STEP_SUMMARY
+        echo "- \`ghcr.io/azure/ai-infrastructure-on-azure/nccl-test:dev2503-${{ github.sha }}\`" >> $GITHUB_STEP_SUMMARY
         echo "" >> $GITHUB_STEP_SUMMARY
         echo "**Changes:**" >> $GITHUB_STEP_SUMMARY
-        echo "- Updated to use PyTorch base image: \`nvcr.io/nvidia/pytorch:25.06-py3\`" >> $GITHUB_STEP_SUMMARY
+        echo "- Updated to use PyTorch base image: \`nvcr.io/nvidia/pytorch:25.03-py3\`" >> $GITHUB_STEP_SUMMARY
         echo "- Includes NCCL tests for distributed training validation" >> $GITHUB_STEP_SUMMARY
         echo "" >> $GITHUB_STEP_SUMMARY
         echo "**Usage:**" >> $GITHUB_STEP_SUMMARY
         echo "\`\`\`bash" >> $GITHUB_STEP_SUMMARY
-        echo "docker pull ghcr.io/azure/ai-infrastructure-on-azure/nccl-test:dev" >> $GITHUB_STEP_SUMMARY
+        echo "docker pull ghcr.io/azure/ai-infrastructure-on-azure/nccl-test:dev2503" >> $GITHUB_STEP_SUMMARY
         echo "\`\`\`" >> $GITHUB_STEP_SUMMARY

infrastructure_validations/aks/NCCL/docker/Dockerfile

@@ -1,4 +1,4 @@
-ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:25.06-py3
+ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:25.03-py3
 FROM ${PYTORCH_IMAGE} AS builder
 
 ENV DEBIAN_FRONTEND=noninteractive
@@ -35,7 +35,7 @@ RUN git clone --branch ${NCCL_TESTS_VERSION} --depth 1 https://github.com/NVIDIA
 RUN git clone --depth=1 https://github.com/Azure/azhpc-images.git /tmp/azhpc-images
 
 # Final minimal image
-FROM nvcr.io/nvidia/pytorch:25.06-py3 AS final
+FROM nvcr.io/nvidia/pytorch:25.03-py3 AS final
 
 ENV DEBIAN_FRONTEND=noninteractive