Use pytorch/pytorch:2.4.1-cpu base image to avoid torch install

Dockerfile

@@ -1,4 +1,4 @@
-FROM python:3.10-slim
+FROM pytorch/pytorch:2.4.1-cpu
 
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PIP_NO_CACHE_DIR=1
@@ -14,55 +14,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 
 WORKDIR /app
 
-# Install Python dependencies in batches to avoid memory issues
+# Install remaining Python dependencies (torch/torchaudio already in base)
 COPY requirements.txt .
-
-# Batch 1: Core ML deps (torch already has index-url)
-RUN pip install --no-cache-dir --timeout=300 \
-    torch==2.4.1+cpu torchaudio==2.4.1+cpu \
-    --index-url https://download.pytorch.org/whl/cpu
-
-# Batch 2: Audio processing
-RUN pip install --no-cache-dir --timeout=300 \
-    numpy>=1.24.0,<2 \
-    scipy>=1.10.0,<2 \
-    librosa>=0.10.0,<0.11 \
-    soundfile>=0.12.0,<1 \
-    ffmpeg-python>=0.2.0,<1 \
-    av>=15.1.0,<16
-
-# Batch 3: ML utilities
-RUN pip install --no-cache-dir --timeout=300 \
-    faiss-cpu>=1.13.0,<2 \
-    praat-parselmouth>=0.4.5,<1 \
-    transformers>=4.49.0,<4.50 \
-    onnxruntime>=1.24.4,<2 \
-    opencv-python-headless>=4.10.0,<5 \
-    scikit-learn>=1.6.0,<2 \
-    einops>=0.8.0,<1
-
-# Batch 4: Web UI
-RUN pip install --no-cache-dir --timeout=300 \
-    gradio>=4.44.0 \
-    fastapi>=0.88.0,<0.100 \
-    uvicorn>=0.20.0,<0.23 \
-    pydantic>=1.10.13,<2 \
-    httpx>=0.23.0,<0.24 \
-    anyio>=3.6.2,<4 \
-    starlette>=0.22.0,<0.28 \
-    websockets>=10.4,<11
-
-# Batch 5: Remaining
-RUN pip install --no-cache-dir --timeout=300 \
-    tqdm>=4.67.0,<5 \
-    huggingface_hub>=0.20.0 \
-    PyYAML>=6.0 \
-    coloredlogs>=15.0,<16 \
-    torchfcpe>=0.0.4,<0.1 \
-    hyper-connections>=0.1.8,<0.4.3 \
-    local-attention>=1.11.0,<2 \
-    networkx>=3.2.0,<4 \
-    matplotlib>=3.8.2,<4
+RUN pip install --no-cache-dir --timeout=300 -r requirements.txt
 
 # Copy RVC source code
 COPY infer ./infer

requirements.txt

@@ -1,9 +1,4 @@
-# Runtime dependencies for RVC inference
-torch==2.4.1+cpu
-torchaudio==2.4.1+cpu
---index-url https://download.pytorch.org/whl/cpu
-
-gradio>=4.44.0
+# Runtime dependencies for RVC inference (torch/torchaudio in base image)
 numpy>=1.24.0,<2
 scipy>=1.10.0,<2
 librosa>=0.10.0,<0.11
@@ -21,4 +16,18 @@ opencv-python-headless>=4.10.0,<5
 scikit-learn>=1.6.0,<2
 PyYAML>=6.0
 coloredlogs>=15.0,<16
-torchfcpe>=0.0.4,<0.1
+torchfcpe>=0.0.4,<0.1
+hyper-connections>=0.1.8,<0.4.3
+local-attention>=1.11.0,<2
+networkx>=3.2.0,<4
+matplotlib>=3.8.2,<4
+
+# Gradio & web deps
+gradio>=4.44.0
+fastapi>=0.88.0,<0.100
+uvicorn>=0.20.0,<0.23
+pydantic>=1.10.13,<2
+httpx>=0.23.0,<0.24
+anyio>=3.6.2,<4
+starlette>=0.22.0,<0.28
+websockets>=10.4,<11