diff --git a/.github/workflows/intermediate-docker.yml b/.github/workflows/intermediate-docker.yml index f158eb4207..fe1f3dde9a 100644 --- a/.github/workflows/intermediate-docker.yml +++ b/.github/workflows/intermediate-docker.yml @@ -86,14 +86,34 @@ jobs: else echo "no BASE_IMAGE match" fi - - name: Build image - id: build + + - name: Build amd64 image + id: build-amd64 + uses: docker/build-push-action@v5 + env: + PLATFORM: amd64 + with: + context: . + push: true + file: build/dockerfiles/intermediate/${{ github.event.inputs.BASE_IMAGE }}.Dockerfile + platforms: linux/amd64 + tags: scrolltech/${{ github.event.inputs.BASE_IMAGE }}:${{ env.TAG }} + build-args: | + CUDA_VERSION=${{ github.event.inputs.CUDA_VERSION }} + GO_VERSION=${{ github.event.inputs.GO_VERSION }} + RUST_VERSION=${{ github.event.inputs.RUST_VERSION }} + PYTHON_VERSION=${{ github.event.inputs.PYTHON_VERSION }} + + - name: Build arm64 image + id: build-arm64 uses: docker/build-push-action@v5 + env: + PLATFORM: arm64 with: context: . push: true file: build/dockerfiles/intermediate/${{ github.event.inputs.BASE_IMAGE }}.Dockerfile - platforms: linux/amd64,linux/arm64 + platforms: linux/arm64 tags: scrolltech/${{ github.event.inputs.BASE_IMAGE }}:${{ env.TAG }} build-args: | CUDA_VERSION=${{ github.event.inputs.CUDA_VERSION }} diff --git a/build/dockerfiles/intermediate/cuda-go-rust-builder.Dockerfile b/build/dockerfiles/intermediate/cuda-go-rust-builder.Dockerfile index 757c7c06a9..e018ae5c55 100644 --- a/build/dockerfiles/intermediate/cuda-go-rust-builder.Dockerfile +++ b/build/dockerfiles/intermediate/cuda-go-rust-builder.Dockerfile @@ -29,7 +29,7 @@ RUN cargo install cargo-chef --locked --version ${CARGO_CHEF_TAG} \ # Install Go ARG GO_VERSION RUN rm -rf /usr/local/go -RUN wget https://go.dev/dl/go${GO_VERSION}.1.linux-amd64.tar.gz -RUN tar -C /usr/local -xzf go${GO_VERSION}.1.linux-amd64.tar.gz -RUN rm go${GO_VERSION}.1.linux-amd64.tar.gz +RUN wget https://go.dev/dl/go${GO_VERSION}.1.linux-${PLATFORM}.tar.gz +RUN tar -C /usr/local -xzf go${GO_VERSION}.1.linux-${PLATFORM}.tar.gz +RUN rm go${GO_VERSION}.1.linux-${PLATFORM}.tar.gz ENV PATH="/usr/local/go/bin:${PATH}" diff --git a/build/dockerfiles/intermediate/go-rust-builder.Dockerfile b/build/dockerfiles/intermediate/go-rust-builder.Dockerfile index 1881a57029..1136fda34f 100644 --- a/build/dockerfiles/intermediate/go-rust-builder.Dockerfile +++ b/build/dockerfiles/intermediate/go-rust-builder.Dockerfile @@ -25,7 +25,7 @@ RUN cargo install cargo-chef --locked --version ${CARGO_CHEF_TAG} \ # Install Go ARG GO_VERSION RUN rm -rf /usr/local/go -RUN wget https://go.dev/dl/go${GO_VERSION}.1.linux-amd64.tar.gz -RUN tar -C /usr/local -xzf go${GO_VERSION}.1.linux-amd64.tar.gz -RUN rm go${GO_VERSION}.1.linux-amd64.tar.gz +RUN wget https://go.dev/dl/go${GO_VERSION}.1.linux-${PLATFORM}.tar.gz +RUN tar -C /usr/local -xzf go${GO_VERSION}.1.linux-${PLATFORM}.tar.gz +RUN rm go${GO_VERSION}.1.linux-${PLATFORM}.tar.gz ENV PATH="/usr/local/go/bin:${PATH}"