Dockerfile

# File: Dockerfile
# Author: Ryoichi Ando (ryoichi.ando@zozo.com)
# License: Apache v2.0

ARG IMAGE
FROM $IMAGE

ARG MODE
ENV NVIDIA_DRIVER_CAPABILITIES=utility,compute
ENV LANG=en_US.UTF-8
ENV PROJ_NAME=ppf-contact-solver
ENV BUILT_MODE=$MODE

COPY . /root/${PROJ_NAME}
WORKDIR /root/${PROJ_NAME}

RUN echo "building in ${MODE} mode"
RUN sleep 5

RUN if [ "$MODE" = "compiled" ]; then \
        /root/.cargo/bin/cargo build --release; \
    elif [ "$MODE" = "base" ]; then \
        apt update; \
        apt install -y git python3 curl; \
        python3 warmup.py; \
        cd /root && rm -rf /root/${PROJ_NAME}; \
    else \
        echo "unknown build mode ${BUILT_MODE}"; \
        exit 1; \
    fi

WORKDIR /root
RUN rm -rf /var/lib/apt/lists/*

CMD if [ "$BUILT_MODE" = "compiled" ]; then \
        cd /root/${PROJ_NAME} && python3 warmup.py jupyter; \
    elif [ "$BUILT_MODE" = "base" ]; then \
        bash; \
    fi