flakego commited on
Commit
352e718
·
verified ·
1 Parent(s): 5337956

Use CPU PyTorch base image for automatic Doubao service

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -2
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM python:3.10-slim
2
 
3
  ENV PYTHONUNBUFFERED=1 \
4
  PIP_NO_CACHE_DIR=1 \
@@ -17,7 +17,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
17
 
18
  COPY requirements.txt .
19
  RUN pip install --upgrade pip \
20
- && pip install torch --index-url https://download.pytorch.org/whl/cpu \
21
  && pip install -r requirements.txt
22
 
23
  COPY app.py doubao_alpha.png ./
 
1
+ FROM pytorch/pytorch:2.3.1-cpu
2
 
3
  ENV PYTHONUNBUFFERED=1 \
4
  PIP_NO_CACHE_DIR=1 \
 
17
 
18
  COPY requirements.txt .
19
  RUN pip install --upgrade pip \
 
20
  && pip install -r requirements.txt
21
 
22
  COPY app.py doubao_alpha.png ./