Gryphus314 commited on
Commit
2e0d3b8
·
verified ·
1 Parent(s): 928fbcb

Update Dockerfile

Browse files

Corrección de fuentes para torch

Files changed (1) hide show
  1. Dockerfile +7 -0
Dockerfile CHANGED
@@ -4,14 +4,21 @@ ENV PYTHONDONTWRITEBYTECODE=1
4
  ENV PYTHONUNBUFFERED=1
5
  ENV DEBIAN_FRONTEND=noninteractive
6
 
 
7
  RUN apt-get update && apt-get install -y --no-install-recommends \
8
  git wget libgl1 libglib2.0-0 libsm6 libxext6 libxrender-dev \
9
  && rm -rf /var/lib/apt/lists/*
10
 
 
11
  RUN useradd -m -u 1000 user
12
  USER user
13
  WORKDIR /app
14
 
 
 
 
 
 
15
  COPY --chown=user:user requirements.txt .
16
  RUN pip install --user --no-cache-dir -r requirements.txt
17
 
 
4
  ENV PYTHONUNBUFFERED=1
5
  ENV DEBIAN_FRONTEND=noninteractive
6
 
7
+ # Instala dependencias sistema
8
  RUN apt-get update && apt-get install -y --no-install-recommends \
9
  git wget libgl1 libglib2.0-0 libsm6 libxext6 libxrender-dev \
10
  && rm -rf /var/lib/apt/lists/*
11
 
12
+ # Crea usuario
13
  RUN useradd -m -u 1000 user
14
  USER user
15
  WORKDIR /app
16
 
17
+ # Instala torch CPU desde index oficial
18
+ RUN pip install --user --no-cache-dir --index-url https://download.pytorch.org/whl/cpu \
19
+ torch torchvision torchaudio
20
+
21
+ # Instala el resto (diffusers ahora encuentra torch compatible)
22
  COPY --chown=user:user requirements.txt .
23
  RUN pip install --user --no-cache-dir -r requirements.txt
24