oiisa commited on
Commit
3958793
·
verified ·
1 Parent(s): 08e254a

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -17,7 +17,7 @@ WORKDIR /app
17
  COPY requirements.txt .
18
 
19
  # Устанавливаем Python зависимости (с совместимым Torch)
20
- RUN pip install --no-cache-dir torch==2.2.0+cpu -f https://download.pytorch.org/whl/cpu
21
  RUN pip install --no-cache-dir -r requirements.txt
22
 
23
  # Кэшируем модель, чтобы не качалась при каждом запуске
 
17
  COPY requirements.txt .
18
 
19
  # Устанавливаем Python зависимости (с совместимым Torch)
20
+ RUN pip install --no-cache-dir torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cpu
21
  RUN pip install --no-cache-dir -r requirements.txt
22
 
23
  # Кэшируем модель, чтобы не качалась при каждом запуске