Update Dockerfile

Dockerfile

@@ -2,16 +2,17 @@
 # MatAnyOne + SAM2 + Gradio Dockerfile (Python 3.10 + CUDA 12.1)
 # ============================================================================
 
-# Use Python 3.10 base
-FROM python:3.10-slim
+# Use NVIDIA CUDA 12.1 base image with Python 3.10
+FROM nvidia/cuda:12.1.0-cudnn8-runtime-ubuntu20.04
 
 # Environment settings
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
     DEBIAN_FRONTEND=noninteractive
 
-# Install system dependencies
+# Install Python 3.10 and system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3.10 python3.10-dev python3-pip python3.10-venv \
     git wget curl build-essential cmake pkg-config \
     ffmpeg libsm6 libxext6 libfontconfig1 libxrender1 \
     libgl1-mesa-glx libglib2.0-0 \
@@ -27,21 +28,21 @@ WORKDIR /app
 RUN useradd -m -u 1000 appuser && chown -R appuser:appuser /app
 USER appuser
 
+# Upgrade pip and install wheel
+RUN python3.10 -m pip install --no-cache-dir --user pip setuptools wheel
+
 # Copy requirements for layer caching
 COPY --chown=appuser:appuser requirements.txt .
 
-# Upgrade pip + core tools
-RUN pip install --no-cache-dir --user pip setuptools wheel
-
 # Install PyTorch (CUDA 12.1 build)
-RUN pip install --no-cache-dir --user \
+RUN python3.10 -m pip install --no-cache-dir --user \
     torch==2.1.0 \
     torchvision==0.16.0 \
     torchaudio==2.1.0 \
     --extra-index-url https://download.pytorch.org/whl/cu121
 
 # Install core CV/scientific dependencies
-RUN pip install --no-cache-dir --user \
+RUN python3.10 -m pip install --no-cache-dir --user \
     numpy==1.26.4 \
     opencv-python-headless==4.10.0.84 \
     Pillow>=10.0.1,<11.0 \
@@ -49,7 +50,7 @@ RUN pip install --no-cache-dir --user \
     av==12.1.0
 
 # Install everything else from requirements.txt
-RUN pip install --no-cache-dir --user -r requirements.txt
+RUN python3.10 -m pip install --no-cache-dir --user -r requirements.txt
 
 # Add user bin to PATH
 ENV PATH="/home/appuser/.local/bin:$PATH"
@@ -75,4 +76,4 @@ HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
 EXPOSE 7860
 
 # Run application
-CMD ["python", "app.py"]
+CMD ["python3.10", "app.py"]