bigmac-pp-simulator / Dockerfile
寅虎
add constraint on micro batch size
164d1e6
Raw
History Blame Contribute Delete
546 Bytes
FROM python:3.11-slim
ENV PYTHONDONTWRITEBYTECODE=1
ENV PYTHONUNBUFFERED=1
ENV MPLCONFIGDIR=/tmp/matplotlib
ENV XDG_CACHE_HOME=/tmp
ENV PP_SIMULATOR_MAX_MICROBATCHES=64
WORKDIR /app
COPY pp-scheduler /app/pp-scheduler
COPY pp-simulator /app/pp-simulator
RUN python -m pip install --no-cache-dir --upgrade pip \
&& python -m pip install --no-cache-dir \
-e /app/pp-scheduler \
-e /app/pp-simulator \
sympy
WORKDIR /app/pp-simulator
EXPOSE 7860
CMD ["python", "web/app.py", "--host", "0.0.0.0", "--port", "7860"]