mirror of
https://github.com/eth-act/ere.git
synced 2026-02-19 11:54:42 -05:00
25 lines
726 B
Docker
25 lines
726 B
Docker
ARG BASE_IMAGE=ere-base:latest
|
|
|
|
FROM $BASE_IMAGE
|
|
|
|
# Set default toolchain to nightly-2025-07-25
|
|
RUN rustup default nightly-2025-07-25
|
|
|
|
# Whether to enable CUDA feature or not.
|
|
ARG CUDA
|
|
|
|
# Env read by Airbender crate `gpu_prover`, numeric cuda arch IDs (e.g. "120" or "89;120")
|
|
ARG CUDAARCHS=120
|
|
|
|
# Copy the Airbender SDK installer script from the workspace context
|
|
COPY --chmod=755 scripts/sdk_installers/install_airbender_sdk.sh /tmp/install_airbender_sdk.sh
|
|
|
|
# Run the Airbender SDK installation script.
|
|
# This script installs airbender-cli.
|
|
RUN /tmp/install_airbender_sdk.sh && rm /tmp/install_airbender_sdk.sh
|
|
|
|
# Verify airbender-cli is accessible with the correct toolchain
|
|
RUN airbender-cli --version
|
|
|
|
CMD ["/bin/bash"]
|