Try to add arm64 variant

hv0905 · Feb 20, 2025 · fb7c3dc · fb7c3dc
1 parent 876e2c0
commit fb7c3dc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/.github/workflows/prod.yml b/.github/workflows/prod.yml
@@ -21,17 +21,20 @@ jobs:
       matrix:
         configurations:
           - dockerfile: "Dockerfile"
+            platforms: "linux/amd64"
             suffixes: |
               ""
               "-cuda"
               "-cuda12.1"
             args: |
               CUDA_VERSION=12.1
           - dockerfile: "Dockerfile"
+            platforms: "linux/amd64"
             suffixes: '"-cuda11.8"'
             args: |
               CUDA_VERSION=11.8
           - dockerfile: "cpu-only.Dockerfile"
+            platforms: "linux/amd64,linux/arm64"
             suffixes: '"-cpu"'
             args: ""
     steps:
@@ -66,6 +69,7 @@ jobs:
         uses: docker/build-push-action@v5
         with:
           file: ${{ matrix.configurations.dockerfile }}
+          platforms: ${{matrix.configurations.platforms}}
           push: true
           tags: ${{ steps.combine-tags.outputs.tags }}
           build-args: ${{ matrix.configurations.args }}