Bumping Cuda version to 12.6

Author: Jack-Khuu

Diff for: .github/workflows/run-readme-pr-linuxaarch64.yml

@@ -12,9 +12,9 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       runner: linux.arm64.m7g.4xlarge
-      docker-image: "pytorch/manylinuxaarch64-builder:cuda12.1-main"
+      docker-image: "pytorch/manylinuxaarch64-builder:cuda12.6-main"
       gpu-arch-type: cuda
-      gpu-arch-version: "12.1"
+      gpu-arch-version: "12.6"
       timeout: 60
       script: |
         echo "::group::Print machine info"
@@ -37,9 +37,9 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       runner: linux.arm64.m7g.4xlarge
-      docker-image: "pytorch/manylinuxaarch64-builder:cuda12.1-main"
+      docker-image: "pytorch/manylinuxaarch64-builder:cuda12.6-main"
       gpu-arch-type: cuda
-      gpu-arch-version: "12.1"
+      gpu-arch-version: "12.6"
       timeout: 60
       script: |
         echo "::group::Print machine info"
@@ -57,9 +57,9 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       runner: linux.arm64.m7g.4xlarge
-      docker-image: "pytorch/manylinuxaarch64-builder:cuda12.1-main"
+      docker-image: "pytorch/manylinuxaarch64-builder:cuda12.6-main"
       gpu-arch-type: cuda
-      gpu-arch-version: "12.1"
+      gpu-arch-version: "12.6"
       timeout: 60
       script: |
         echo "::group::Print machine info"
@@ -82,9 +82,9 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       runner: linux.arm64.m7g.4xlarge
-      docker-image: "pytorch/manylinuxaarch64-builder:cuda12.1-main"
+      docker-image: "pytorch/manylinuxaarch64-builder:cuda12.6-main"
       gpu-arch-type: cuda
-      gpu-arch-version: "12.1"
+      gpu-arch-version: "12.6"
       timeout: 60
       script: |
         echo "::group::Print machine info"
@@ -107,9 +107,9 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       runner: linux.arm64.m7g.4xlarge
-      docker-image: "pytorch/manylinuxaarch64-builder:cuda12.1-main"
+      docker-image: "pytorch/manylinuxaarch64-builder:cuda12.6-main"
       gpu-arch-type: cuda
-      gpu-arch-version: "12.1"
+      gpu-arch-version: "12.6"
       timeout: 60
       script: |
         echo "::group::Print machine info"