Update Dockerfile

Dockerfile

@@ -1,33 +1,34 @@
-FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
+FROM nvidia/cuda:12.1.105-cudnn8-devel-ubuntu22.04
 
-SHELL ["/bin/bash", "-c"]
-ENV PATH=/opt/conda/bin:$PATH \
-    HF_HOME=/app/hf_cache \
-    TRANSFORMERS_CACHE=/app/hf_cache \
-    PIP_NO_CACHE_DIR=1 \
-    PYTHONDONTWRITEBYTECODE=1 \
-    PYTHONUNBUFFERED=1
+ENV DEBIAN_FRONTEND=noninteractive \
+    PATH=/opt/conda/bin:$PATH \
+    CUDA_HOME=/usr/local/cuda
 
-WORKDIR /app
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    wget git libgl1-mesa-glx libgl1-mesa-dri xvfb ffmpeg build-essential \
+    && rm -rf /var/lib/apt/lists/*
 
-# OS deps (one layer) + clean
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    git wget curl unzip ffmpeg libgl1 libglib2.0-0 \
- && rm -rf /var/lib/apt/lists/*
+# Install Miniconda
+RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /tmp/miniconda.sh && \
+    bash /tmp/miniconda.sh -b -p /opt/conda && \
+    rm /tmp/miniconda.sh && \
+    /opt/conda/bin/conda clean -afy
 
-# Leverage build cache: install Python deps first
-COPY requirements.txt /app/requirements.txt
-RUN pip install --upgrade pip && pip install -r /app/requirements.txt
+# Create conda env
+RUN conda create -n appenv python=3.10 -y && conda clean -afy
+SHELL ["conda", "run", "-n", "appenv", "/bin/bash", "-c"]
 
-# Then copy the rest of the project
-COPY . /app
+# Install ffmpeg from conda-forge
+RUN conda install ffmpeg=7 -c conda-forge -y && conda clean -afy
 
-# Optional: run as non-root (better security)
-# RUN useradd -m app && chown -R app:app /app
-# USER app
+# Copy project
+WORKDIR /workspace
+COPY requirements.txt .
 
-# Gradio on 0.0.0.0 in containers
-ENV GRADIO_SERVER_NAME=0.0.0.0
-EXPOSE 7860
+# Install Python dependencies (flash-attn will now compile)
+RUN pip install --upgrade pip && pip install --no-cache-dir -r requirements.txt
 
-CMD ["python", "finetune/gradio_app.py"]
+COPY . .
+
+CMD ["conda", "run", "--no-capture-output", "-n", "appenv", "python", "app.py"]