luisBJ/super-resolution-model

1FROM nvidia/cuda:12.2.0-base-ubuntu22.042 3WORKDIR /app4 5# Instalar dependencias del sistema6RUN apt-get update && \7    apt-get install -y --no-install-recommends \8    python3.10 \9    python3-pip \10    libgl1 \11    libglib2.0-0 && \12    rm -rf /var/lib/apt/lists/*13 14# Copiar todo el proyecto15COPY . .16 17# Instalar dependencias Python18# Instalar PyTorch sin soporte CUDA19RUN pip install --no-cache-dir torch==2.1.0+cpu torchvision==0.16.0+cpu -f https://download.pytorch.org/whl/torch_stable.html20 21# Luego instala el resto de dependencias22RUN pip install --no-cache-dir -r requirements.txt23RUN pip install --no-cache-dir uvicorn24 25# Crear directorios permitidos si no existen26RUN mkdir -p /outputs/visualization/Inference27 28# Variables de entorno para configuraciones29ENV PYTHONPATH=/app30ENV HF_MODEL_PATH=/app/experiments/trained_models31# Asegúrate de que el nombre y extensión concuerden con lo usado en app.py32ENV CONFIG_PATH=/app/options/test/HMA_SRx2.yml33ENV PYTHONUNBUFFERED=134ENV CUDA_VISIBLE_DEVICES=-135ENV HF_HOME=/cache/huggingface36 37EXPOSE 786038 39CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]40