diff --git a/flask_ml_api/Dockerfile b/flask_ml_api/Dockerfile index 49dd2be619a905bbaaa8c04883e9e478095245ba..e69265fae46ac6ffea56577a965d0fc1810392ad 100755 --- a/flask_ml_api/Dockerfile +++ b/flask_ml_api/Dockerfile @@ -8,10 +8,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ RUN python3 -m venv /opt/myenv ENV PATH="/opt/myenv/bin:$PATH" -#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 && \ @@ -20,6 +16,8 @@ RUN pip install --no-cache-dir Flask && \ pip install --no-cache-dir sentence_transformers && \ pip install --no-cache-dir --upgrade setuptools wheel +RUN docker system prune -a --volumes + FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime COPY --from=build /opt/myenv /opt/myenv