49 lines
1.4 KiB
Docker
49 lines
1.4 KiB
Docker
FROM python:3.11-slim
|
|
|
|
ARG USER=app
|
|
ARG UID=1000
|
|
ENV APP_HOME=/app \
|
|
PYTHONDONTWRITEBYTECODE=1 \
|
|
PYTHONUNBUFFERED=1 \
|
|
PIP_NO_CACHE_DIR=1 \
|
|
PORT=8000
|
|
|
|
RUN sed -i 's/^\(Components:.*\)$/\1 contrib non-free/' /etc/apt/sources.list.d/debian.sources
|
|
|
|
RUN apt-get update \
|
|
&& apt-get install -y --no-install-recommends \
|
|
gnupg2 \
|
|
curl \
|
|
ca-certificates \
|
|
lsb-release \
|
|
wget \
|
|
apt-transport-https \
|
|
nvidia-smi \
|
|
procps \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
RUN wget https://repo.radeon.com/rocm/rocm.gpg.key -O - | gpg --dearmor | tee /etc/apt/keyrings/rocm.gpg > /dev/null \
|
|
&& echo "deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/rocm/apt/7.1 noble main" | tee /etc/apt/sources.list.d/rocm.list \
|
|
&& apt-get update \
|
|
&& apt-get install -y --no-install-recommends \
|
|
rocm-smi \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
RUN useradd --create-home --uid ${UID} ${USER}
|
|
|
|
WORKDIR ${APP_HOME}
|
|
|
|
COPY --chown=${USER}:${USER} requirements.txt ${APP_HOME}/
|
|
RUN pip install --upgrade pip setuptools wheel \
|
|
&& pip install --no-cache-dir -r requirements.txt
|
|
|
|
COPY --chown=${USER}:${USER} . ${APP_HOME}/
|
|
|
|
USER ${USER}
|
|
|
|
EXPOSE ${PORT}
|
|
|
|
HEALTHCHECK --interval=30s --timeout=3s --start-period=10s --retries=3 \
|
|
CMD curl -f http://localhost:${PORT}/metrics || exit 1
|
|
|
|
CMD ["python", "-u", "app.py"] |