222

Dockerfile

@@ -1,38 +1,30 @@
 # Usa una imagen base con soporte para Python y CUDA
 FROM nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu20.04
 
-# Establece variables de entorno necesarias para el caché y CUDA
-ENV TRANSFORMERS_CACHE="/app/.cache" \
-    HF_HOME="/app/.cache" \
-    PATH="/opt/conda/bin:$PATH" \
-    DEBIAN_FRONTEND=noninteractive
-
-# Actualiza e instala dependencias básicas
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    python3 \
-    python3-pip \
-    git \
-    && apt-get clean \
-    && rm -rf /var/lib/apt/lists/*
+# Instala Miniconda
+RUN apt-get update && apt-get install -y wget && \
+    wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
+    bash Miniconda3-latest-Linux-x86_64.sh -b -p /opt/conda && \
+    rm Miniconda3-latest-Linux-x86_64.sh && \
+    apt-get clean
 
-# Establece el directorio de trabajo
-WORKDIR /app
+# Configura Conda en el PATH
+ENV PATH=/opt/conda/bin:$PATH
 
-# Crea el directorio de caché y asigna permisos
-RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache
+# Copia el archivo environment.yaml al contenedor
+COPY environment.yaml /app/environment.yaml
 
-# Copia el archivo requirements.txt al contenedor
-COPY requirements.txt /app/requirements.txt
+# Instala el entorno Conda
+RUN conda env create -f /app/environment.yaml
 
-# Actualiza pip y las dependencias de Python
-RUN python3 -m pip install --upgrade pip
-RUN pip install -r requirements.txt
+# Configura Conda para ejecutarse automáticamente
+SHELL ["conda", "run", "-n", "marigold", "/bin/bash", "-c"]
 
-# Copia todos los archivos de la aplicación al contenedor
-COPY . /app
+# Establece el directorio de trabajo
+WORKDIR /app
 
-# Expone el puerto donde la aplicación se ejecutará (opcional, si es necesario)
-EXPOSE 7860
+# Copia los archivos de la aplicación
+COPY . /app
 
 # Comando para ejecutar la aplicación
-CMD ["python3", "app.py"]
+CMD ["conda", "run", "--no-capture-output", "-n", "marigold", "python", "app.py"]

environment.yaml

@@ -0,0 +1,16 @@
+name: marigold
+channels:
+  - conda-forge
+  - nvidia
+  - pytorch
+dependencies:
+  - accelerate>=0.22.0
+  - diffusers>=0.25.0
+  - matplotlib>=3.8.2
+  - scipy>=1.11.4
+  - python=3.10.12
+  - pytorch=2.4.0
+  - pytorch-cuda=11.7
+  - torchvision=0.19.0
+  - pip:
+    - transformers>=4.32.1