Update Dockerfile

Dockerfile

@@ -8,20 +8,37 @@ ENV PYTHONUNBUFFERED=1 \
     PATH=/usr/local/cuda/bin:$PATH \
     LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
 
-# Install Python and pip
+# Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     python3 \
     python3-pip \
+    python3-dev \
+    build-essential \
     curl \
+    ffmpeg \
+    libsndfile1 \
     && rm -rf /var/lib/apt/lists/*
 
+# Upgrade pip and install build tools
+RUN python3 -m pip install --no-cache-dir --upgrade pip setuptools wheel
+
+# Install PyTorch separately first
+RUN pip3 install --no-cache-dir \
+    torch==2.4.1 \
+    torchaudio==2.4.1 \
+    torchvision==0.19.1 \
+    --extra-index-url https://download.pytorch.org/whl/cu121
+
 # Set up working directory
 WORKDIR /app
 
 # Copy requirements first to leverage Docker cache
 COPY requirements.txt .
 
-# Install Python dependencies
+# Modify requirements to remove PyTorch packages
+RUN sed -i '/^torch==/d; /^torchaudio==/d; /^torchvision==/d; /^torchdata==/d' requirements.txt
+
+# Install remaining Python dependencies
 RUN pip3 install --no-cache-dir -r requirements.txt
 
 # Copy the rest of your application