Przeglądaj źródła

fix(server): Flask not found (#1830)

* fix(server): Flask not found

* trial 1
Alex 2 lat temu
rodzic
commit
a3b9a0be3a
1 zmienionych plików z 8 dodań i 3 usunięć
  1. 8 3
      machine-learning/Dockerfile

+ 8 - 3
machine-learning/Dockerfile

@@ -1,11 +1,16 @@
 FROM python:3.10
 
 ENV TRANSFORMERS_CACHE=/cache
+ENV PYTHONDONTWRITEBYTECODE 1
+ENV PYTHONUNBUFFERED 1
 
 WORKDIR /usr/src/app
 
-RUN pip install --user --no-cache-dir torch==1.13.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
-RUN pip install --user transformers tqdm numpy scikit-learn scipy nltk sentencepiece flask Pillow
-RUN pip install --user --no-deps sentence-transformers
+RUN python -m venv /opt/venv
+ENV PATH="/opt/venv/bin:$PATH"
+
+RUN pip install --no-cache-dir torch==1.13.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
+RUN pip install transformers tqdm numpy scikit-learn scipy nltk sentencepiece flask Pillow
+RUN pip install --no-deps sentence-transformers
 
 COPY . .