Update Dockerfile

Dockerfile

@@ -22,21 +22,22 @@ RUN sed -i 's/main/main contrib non-free/' /etc/apt/sources.list && \
     libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-# Install build dependencies (needed for some Python packages)
+# Install build dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
     python3-dev \
-    python3-pip \
     git \
     && rm -rf /var/lib/apt/lists/*
 
 # Upgrade pip first
 RUN pip install --no-cache-dir --upgrade pip
 
-# Copy and install requirements first for better caching
+# Install PyTorch with CUDA 11.8 support first
+RUN pip install --no-cache-dir torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu118
+
+# Copy and install remaining requirements
 COPY requirements.txt .
-RUN pip install --no-cache-dir packaging && \
-    pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy app code
 COPY . .