Merge pull request #552 from kroma-network/feat/add-circom-docker

feat: add Dockerfile for circom

docker/Dockerfile.circom.jammy

@@ -0,0 +1,45 @@
+FROM zktachyon/base:jammy-latest AS builder
+LABEL maintainer="The Tachyon Authors <[email protected]>"
+
+# Install python dependencies
+RUN apt update && apt install -y --no-install-recommends python3-pip
+RUN python3 -m pip install numpy
+
+# Set the working directory
+WORKDIR /tachyon
+
+# Copy the entire project
+COPY . .
+
+# Build circom prover
+RUN cd vendors/circom && CARGO_BAZEL_REPIN=true bazel build --@kroma_network_tachyon//:has_openmp --config maxopt //:prover_main
+
+# Build compile tool for witness generator
+RUN cd vendors/circom && bazel build --config opt //circomlib/build:compile_witness_generator
+
+# Use the CUDA base image
+FROM nvidia/cuda:12.5.0-devel-ubuntu22.04 AS builder-cuda
+
+# Set the working directory
+WORKDIR /tachyon
+
+# Copy the entire project
+COPY . .
+
+COPY --from=builder /usr /usr
+COPY --from=builder /root/.pyenv /root/.pyenv
+
+RUN echo 'export PYENV_ROOT="/root/.pyenv"' >> ~/.bash_profile && \
+    echo '[[ -d $PYENV_ROOT/bin ]] && export PATH="$PYENV_ROOT/bin:$PATH"' >> ~/.bash_profile && \
+    echo 'eval "$(pyenv init -)"' >> ~/.bash_profile
+RUN echo 'export PYENV_ROOT="/root/.pyenv"' >> ~/.bashrc && \
+    echo 'command -v pyenv >/dev/null || export PATH="$PYENV_ROOT/bin:$PATH"' >> ~/.bashrc && \
+    echo 'eval "$(pyenv init -)"' >> ~/.bashrc
+ENV PYENV_ROOT="/root/.pyenv"
+ENV PATH="${PYENV_ROOT}/shims:${PYENV_ROOT}/bin:${PATH}"
+
+# Build circom prover with CUDA acceleration
+RUN cd vendors/circom && CARGO_BAZEL_REPIN=true bazel build --@kroma_network_tachyon//:has_openmp --config maxopt --config cuda //:prover_main
+
+# Build compile tool for witness generator
+RUN cd vendors/circom && bazel build --config opt //circomlib/build:compile_witness_generator

vendors/circom/README.md

@@ -43,6 +43,23 @@ bazel build --@kroma_network_tachyon//:has_openmp --config maxopt --config cuda
 bazel build --@kroma_network_tachyon//:has_openmp --config maxopt //:prover_main
 ```
 
+### With Docker
+
+For the CUDA build, you need to [setup the NVIDIA Container toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html).
+
+```shell
+# These commands should be run in the base folder
+cd ../..
+# Build without CUDA
+docker build -f docker/Dockerfile.circom.jammy -t tachyon:cpu  --target builder .
+docker run -it tachyon:cpu
+# Build with CUDA
+docker build -f docker/Dockerfile.circom.jammy -t tachyon:cuda  --target builder-cuda .
+docker run -it --runtime nvidia --gpus all tachyon:cuda
+```
+
+NOTE: To use external files for proving or compiling, you need to mount or copy them to the tachyon folder to use it within docker.
+
 ## How to run
 
 ```shell