25 lines
571 B
Docker
25 lines
571 B
Docker
FROM pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime
|
|
|
|
ENV PYTHONDONTWRITEBYTECODE=1
|
|
ENV PYTHONUNBUFFERED=1
|
|
|
|
WORKDIR /app
|
|
|
|
# System dependencies for OpenCV and image processing
|
|
RUN apt-get update && apt-get install -y \
|
|
libglib2.0-0 \
|
|
libsm6 \
|
|
libxext6 \
|
|
libxrender-dev \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Install python packages
|
|
COPY requirements.txt .
|
|
RUN pip install --no-cache-dir -r requirements.txt
|
|
|
|
# Copy source code
|
|
COPY src/ /app/src/
|
|
|
|
EXPOSE 8080
|
|
|
|
CMD ["streamlit", "run", "src/main.py", "--server.port", "8080", "--server.address", "0.0.0.0"] |