mirror of
https://github.com/eth-act/ere.git
synced 2026-02-19 11:54:42 -05:00
35 lines
915 B
Docker
35 lines
915 B
Docker
ARG BASE_ZKVM_IMAGE=ere-base-airbender:latest
|
|
ARG RUNTIME_IMAGE=ubuntu:24.04
|
|
ARG RUNTIME_CUDA_IMAGE=nvidia/cuda:12.9.1-runtime-ubuntu24.04
|
|
|
|
# Whether to enable CUDA feature or not.
|
|
ARG CUDA
|
|
|
|
FROM $BASE_ZKVM_IMAGE AS build_stage
|
|
|
|
COPY . /ere
|
|
|
|
WORKDIR /ere
|
|
|
|
ARG CUDA
|
|
ARG RUSTFLAGS
|
|
|
|
RUN cargo build --release --package ere-server --bin ere-server --features airbender${CUDA:+,cuda} \
|
|
&& mkdir bin && mv target/release/ere-server bin/ere-server \
|
|
&& cargo clean && rm -rf $CARGO_HOME/registry/
|
|
|
|
FROM $RUNTIME_IMAGE AS runtime
|
|
FROM $RUNTIME_CUDA_IMAGE AS runtime_cuda
|
|
FROM runtime${CUDA:+_cuda} AS runtime_stage
|
|
|
|
# Copy airbender-cli
|
|
COPY --from=build_stage /usr/local/cargo/bin/airbender-cli /usr/local/cargo/bin/airbender-cli
|
|
|
|
# Add airbender-cli to path
|
|
ENV PATH=/usr/local/cargo/bin:$PATH
|
|
|
|
# Copy ere-server
|
|
COPY --from=build_stage /ere/bin/ere-server /ere/bin/ere-server
|
|
|
|
ENTRYPOINT ["/ere/bin/ere-server"]
|