Use amazon linux2 for Docker builds, fix build-docker-conda condition (#134116)
1. Switches failing jobs to amzon linux 2:
- CUDA, CPU, ROCM jobs are failing
3. Fix trigger condition for build-docker-conda to be same as manywheel and libtorch
Pull Request resolved: https://github.com/pytorch/pytorch/pull/134116
Approved by: https://github.com/ZainRizvi, https://github.com/nWEIdia
diff --git a/.github/actionlint.yaml b/.github/actionlint.yaml
index db4fd84..3fdb47c 100644
--- a/.github/actionlint.yaml
+++ b/.github/actionlint.yaml
@@ -10,6 +10,7 @@
- linux.2xlarge
- linux.4xlarge
- linux.9xlarge.ephemeral
+ - am2.linux.9xlarge.ephemeral
- linux.12xlarge
- linux.12xlarge.ephemeral
- linux.24xlarge
diff --git a/.github/workflows/build-conda-images.yml b/.github/workflows/build-conda-images.yml
index bc687df..4962276 100644
--- a/.github/workflows/build-conda-images.yml
+++ b/.github/workflows/build-conda-images.yml
@@ -11,13 +11,13 @@
# Release candidate tags look like: v1.11.0-rc1
- v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
paths:
- - conda/Dockerfile
- - 'common/*'
+ - '.ci/docker/conda/*'
+ - '.ci/docker/common/*'
- .github/workflows/build-conda-images.yml
pull_request:
paths:
- - conda/Dockerfile
- - 'common/*'
+ - '.ci/docker/conda/*'
+ - '.ci/docker/common/*'
- .github/workflows/build-conda-images.yml
env:
@@ -32,7 +32,7 @@
jobs:
build-docker:
environment: ${{ (github.ref == 'refs/heads/main' || startsWith(github.event.ref, 'refs/tags/v')) && 'docker-build' || '' }}
- runs-on: linux.9xlarge.ephemeral
+ runs-on: am2.linux.9xlarge.ephemeral
strategy:
matrix:
cuda_version: ["11.8", "12.1", "12.4", "cpu"]
diff --git a/.github/workflows/build-manywheel-images.yml b/.github/workflows/build-manywheel-images.yml
index 49beaf3..a85d0c5 100644
--- a/.github/workflows/build-manywheel-images.yml
+++ b/.github/workflows/build-manywheel-images.yml
@@ -33,7 +33,7 @@
jobs:
build-docker-cuda:
environment: ${{ (github.ref == 'refs/heads/main' || startsWith(github.event.ref, 'refs/tags/v')) && 'docker-build' || '' }}
- runs-on: linux.9xlarge.ephemeral
+ runs-on: am2.linux.9xlarge.ephemeral
strategy:
matrix:
cuda_version: ["12.4", "12.1", "11.8"]
@@ -141,7 +141,7 @@
.ci/docker/manywheel/build.sh manylinuxaarch64-builder:cuda${{matrix.cuda_version}}
build-docker-rocm:
environment: ${{ (github.ref == 'refs/heads/main' || startsWith(github.event.ref, 'refs/tags/v')) && 'docker-build' || '' }}
- runs-on: linux.9xlarge.ephemeral
+ runs-on: am2.linux.9xlarge.ephemeral
strategy:
matrix:
rocm_version: ["6.1", "6.2"]
@@ -176,7 +176,7 @@
.ci/docker/manywheel/build.sh manylinux-builder:rocm${{matrix.rocm_version}}
build-docker-cpu:
environment: ${{ (github.ref == 'refs/heads/main' || startsWith(github.event.ref, 'refs/tags/v')) && 'docker-build' || '' }}
- runs-on: linux.9xlarge.ephemeral
+ runs-on: am2.linux.9xlarge.ephemeral
steps:
- name: Checkout PyTorch
uses: pytorch/pytorch/.github/actions/checkout-pytorch@main
