J

VideoBackgroundReplacer2/Dockerfile

@@ -4,6 +4,8 @@
 # SAM2 installed from source; MatAnyone via pip (repo)
 # ===============================
 
+# Set Python version explicitly
+ARG PYTHON_VERSION=3.10
 FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04
 
 # ---- Environment (runtime hygiene) ----
@@ -24,6 +26,11 @@ ENV DEBIAN_FRONTEND=noninteractive \
     STREAMLIT_SERVER_HEADLESS=true \
     STREAMLIT_BROWSER_GATHER_USAGE_STATS=false
 
+# ---- System cleanup ----
+RUN rm -rf /var/lib/apt/lists/* && \
+    apt-get clean && \
+    rm -rf /tmp/* /var/tmp/*
+
 # ---- Non-root user ----
 RUN useradd -m -u 1000 user
 ENV HOME=/home/user
@@ -32,14 +39,17 @@ WORKDIR $HOME/app
 # ---- System deps ----
 RUN apt-get update && apt-get install -y --no-install-recommends \
     git ffmpeg wget curl \
-    python3 python3-pip python3-venv python3-dev \
+    python${PYTHON_VERSION} python${PYTHON_VERSION}-dev python3-pip python${PYTHON_VERSION}-venv \
     build-essential gcc g++ pkg-config \
     libffi-dev libssl-dev libc6-dev \
     libgl1-mesa-glx libglib2.0-0 libsm6 libxext6 libxrender1 libgomp1 \
+    && update-alternatives --install /usr/bin/python3 python3 /usr/bin/python${PYTHON_VERSION} 1 \
+    && update-alternatives --set python3 /usr/bin/python${PYTHON_VERSION} \
+    && ln -sf /usr/bin/python${PYTHON_VERSION} /usr/bin/python \
     && rm -rf /var/lib/apt/lists/*
 
 # ---- Python bootstrap ----
-RUN python3 -m pip install --upgrade pip setuptools wheel
+RUN python3 -m pip install --no-cache-dir --upgrade pip setuptools wheel
 
 # ---- Install PyTorch (CUDA 12.1 wheels) ----
 RUN python3 -m pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cu121 \
@@ -54,8 +64,9 @@ PY
 # ---- Copy deps first (better caching) ----
 COPY --chown=user:user requirements.txt ./
 
-# ---- Install remaining Python deps ----
-RUN python3 -m pip install --no-cache-dir -r requirements.txt
+# ---- Install remaining Python deps with force reinstall ----
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt --force-reinstall
 
 # ---- MatAnyone (pip install from repo with retry) ----
 RUN echo "Installing MatAnyone..." && \
@@ -112,6 +123,10 @@ CMD ["sh", "-c", "\
   echo 'Files in app directory:' && \
   ls -la && \
   echo '' && \
+  echo '=== PYTHON ENVIRONMENT ===' && \
+  python3 --version && \
+  pip --version && \
+  echo '' && \
   echo '=== APP.PY VERIFICATION ===' && \
   if [ -f app.py ]; then \
       echo '✅ app.py found' && \