diff --git a/official-templates/base/docker-bake.hcl b/official-templates/base/docker-bake.hcl index 0fb8a4f4..e21afcf0 100644 --- a/official-templates/base/docker-bake.hcl +++ b/official-templates/base/docker-bake.hcl @@ -9,7 +9,7 @@ group "default" { target "common-base" { context = "official-templates/base" dockerfile = "Dockerfile" - platforms = ["linux/amd64"] + platforms = ["linux/amd64", "linux/arm64"] contexts = { scripts = "container-template" proxy = "container-template/proxy" diff --git a/official-templates/pytorch/docker-bake.hcl b/official-templates/pytorch/docker-bake.hcl index ca5cfc93..bf97b924 100644 --- a/official-templates/pytorch/docker-bake.hcl +++ b/official-templates/pytorch/docker-bake.hcl @@ -79,7 +79,7 @@ group "default" { target "pytorch-base" { context = "official-templates/pytorch" dockerfile = "Dockerfile" - platforms = ["linux/amd64"] + platforms = ["linux/amd64", "linux/arm64"] } target "pytorch-matrix" { diff --git a/official-templates/rocm/docker-bake.hcl b/official-templates/rocm/docker-bake.hcl index 7de08913..034593c6 100644 --- a/official-templates/rocm/docker-bake.hcl +++ b/official-templates/rocm/docker-bake.hcl @@ -10,7 +10,7 @@ group "default" { target "rocm-base" { context = "official-templates/base" dockerfile = "Dockerfile" - platforms = ["linux/amd64"] + platforms = ["linux/amd64", "linux/arm64"] contexts = { scripts = "container-template" proxy = "container-template/proxy"