CRITICAL FIX: Use correct Dockerfile and cleaned requirements.txt

Dockerfile

@@ -1,33 +1,31 @@
-# Use a standard Python 3.12 base image
-FROM python:3.12-slim
+# Use an NVIDIA CUDA runtime as the base image for GPU support
+FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 
-# Set the working directory inside the container
+# Avoid prompts during package installation
+ENV DEBIAN_FRONTEND=noninteractive
+
+# Install system dependencies: Python, pip, git, and git-lfs
+RUN apt-get update && \
+    apt-get install -y python3.11 python3-pip git git-lfs && \
+    rm -rf /var/lib/apt/lists/*
+
+# Set the working directory
 WORKDIR /code
 
-# Install necessary system packages.
-# This is a comprehensive list to satisfy all common headless OpenCV dependencies.
-RUN apt-get update && apt-get install -y \
-    git \
-    wget \
-    aria2 \
-    libgl1-mesa-glx \
-    libglib2.0-0 \
-    libsm6 \
-    libxext6 \
-    && rm -rf /var/lib/apt/lists/*
-
-# Copy the requirements file first to leverage Docker cache
-COPY requirements.txt .
-
-# Install the correct CUDA-enabled PyTorch version and other requirements
-RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Copy all project files, including the start.sh script
+# Install the correct GPU-enabled PyTorch version for CUDA 12.1
+RUN pip3 install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 --index-url https://download.pytorch.org/whl/cu121
+
+# Copy your cleaned requirements file
+COPY requirements.txt requirements.txt
+
+# Install all other Python dependencies
+RUN pip3 install --no-cache-dir -r requirements.txt
+
+# Copy the rest of your application code
 COPY . .
 
-# Make the startup script executable
-RUN chmod +x start.sh
+# Make the start script executable
+RUN chmod +x ./start.sh
 
-# The command is just to run the script.
-CMD ["./start.sh"]
+# Run start.sh when the container launches
+CMD ["/bin/bash", "./start.sh"]

requirements.txt

@@ -1,5 +1,4 @@
-torch
-torchvision
+# requirements.txt
 accelerate==1.2.1
 av==14.0.1
 bitsandbytes==0.45.0
@@ -13,12 +12,5 @@ toml==0.10.2
 tqdm==4.67.1
 transformers==4.46.3
 voluptuous==0.15.2
-
-# Wan2.1
 ftfy==6.3.1
-easydict==1.13
-
-# optional dependencies
-# ascii-magic==2.3.0
-# matplotlib==3.10.0
-# tensorboard
+easydict==1.13