diff --git a/.github/workflows/run-readme-pr-linuxaarch64.yml b/.github/workflows/run-readme-pr-linuxaarch64.yml index 6db867df3..842d1f66b 100644 --- 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