File size: 667 Bytes
fb6a37c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | FROM python:3.10-slim
RUN apt-get update && apt-get install -y --no-install-recommends \
libgl1 libglib2.0-0 gcc g++ \
&& rm -rf /var/lib/apt/lists/*
RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user PATH=/home/user/.local/bin:$PATH
WORKDIR $HOME/app
# Install CPU-only PyTorch first (critical: use --index-url not --extra-index-url)
RUN pip install --no-cache-dir \
torch==2.1.0+cpu torchvision==0.16.0+cpu \
--index-url https://download.pytorch.org/whl/cpu
# Install remaining deps
COPY --chown=user requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
COPY --chown=user app.py .
EXPOSE 7860
CMD ["python", "app.py"]
|