diff --git a/flask_ml_api/Dockerfile b/flask_ml_api/Dockerfile index 0c49c58f6d57e4253dc474eeda8c41d332281522..8944451858c40cb1c009eb8137bf67a7e4b4c7c1 100755 --- a/flask_ml_api/Dockerfile +++ b/flask_ml_api/Dockerfile @@ -12,15 +12,18 @@ ENV PATH="/opt/myenv/bin:$PATH" WORKDIR /app COPY . /app -RUN pip install --no-cache-dir Flask && \ - pip install --no-cache-dir matplotlib && \ - pip install --no-cache-dir scikit-learn && \ - pip install --no-cache-dir transformers && \ - pip install --no-cache-dir pandas && \ - pip install --no-cache-dir sentence_transformers && \ - pip install --no-cache-dir --upgrade setuptools wheel - -RUN rm -rf /root/.cache/pip +RUN pip install --no-cache-dir Flask matplotlib scikit-learn transformers pandas sentence_transformers && \ + pip install --no-cache-dir --upgrade setuptools wheel && \ + rm -rf /root/.cache/pip \ + +#RUN pip install --no-cache-dir Flask && \ +# pip install --no-cache-dir matplotlib && \ +# pip install --no-cache-dir scikit-learn && \ +# pip install --no-cache-dir transformers && \ +# pip install --no-cache-dir pandas && \ +# pip install --no-cache-dir sentence_transformers && \ +# pip install --no-cache-dir --upgrade setuptools wheel +#RUN rm -rf /root/.cache/pip FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime