Explicitly overriding the docker-image

pytorch · Jan 23, 2025 · a9bb581 · a9bb581
1 parent d3a4673
commit a9bb581
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/.github/workflows/run-readme-pr-linuxaarch64.yml b/.github/workflows/run-readme-pr-linuxaarch64.yml
@@ -12,6 +12,7 @@ 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"
       gpu-arch-type: cuda
       gpu-arch-version: "12.1"
       timeout: 60
@@ -36,6 +37,7 @@ 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"
       gpu-arch-type: cuda
       gpu-arch-version: "12.1"
       timeout: 60
@@ -55,6 +57,7 @@ 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"
       gpu-arch-type: cuda
       gpu-arch-version: "12.1"
       timeout: 60
@@ -79,6 +82,7 @@ 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"
       gpu-arch-type: cuda
       gpu-arch-version: "12.1"
       timeout: 60
@@ -103,6 +107,7 @@ 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"
       gpu-arch-type: cuda
       gpu-arch-version: "12.1"
       timeout: 60