Update Dockerfile

Dockerfile

@@ -1,5 +1,4 @@
 FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
-
 ENV DEBIAN_FRONTEND=noninteractive \
     PATH=/opt/conda/bin:$PATH \
     CUDA_HOME=/usr/local/cuda
@@ -13,22 +12,25 @@ RUN apt-get update && apt-get install -y \
 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
+    conda clean -afy
 
-# Create conda env
-RUN conda create -n appenv python=3.10 -y && conda clean -afy
-SHELL ["conda", "run", "-n", "appenv", "/bin/bash", "-c"]
+# Configure conda to avoid Anaconda TOS prompt
+RUN conda config --remove channels defaults && \
+    conda config --add channels conda-forge && \
+    conda config --set channel_priority strict
 
-# Install ffmpeg from conda-forge
-RUN conda install ffmpeg=7 -c conda-forge -y && conda clean -afy
+# Create conda environment with Python + ffmpeg from conda-forge
+RUN conda create -n appenv python=3.10 ffmpeg=7 -y && conda clean -afy
+SHELL ["conda", "run", "-n", "appenv", "/bin/bash", "-c"]
 
-# Copy project
+# Copy requirements first for layer caching
 WORKDIR /workspace
 COPY requirements.txt .
 
-# Install Python dependencies (flash-attn will now compile)
+# Install Python dependencies (flash-attn will compile since we have nvcc)
 RUN pip install --upgrade pip && pip install --no-cache-dir -r requirements.txt
 
+# Copy remaining code
 COPY . .
 
 CMD ["conda", "run", "--no-capture-output", "-n", "appenv", "python", "finetune/app.py"]