Image build is optimized.

940e64b7 · Artem Baranovskyi · 2e158f9a · 940e64b7
Commit 940e64b7 authored 4 months ago by Artem Baranovskyi
Hide whitespace changes
Inline Side-by-side

Showing

with 2 additions and 2 deletions
+2 -2
--- a/flask_ml_api/Dockerfile
+++ b/flask_ml_api/Dockerfile
 #FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime as build
-FROM pytorch/pytorch:1.10.0-cuda11.7-cudnn8-runtime as build
+FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime as build

 RUN apt-get update && apt-get install -y --no-install-recommends \
    build-essential python3-venv && \
@@ -23,7 +23,7 @@ RUN pip install --no-cache-dir Flask && \
 RUN rm -rf /root/.cache/pip


-FROM pytorch/pytorch:1.10.0-cuda11.7-cudnn8-runtime
+FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime
 COPY --from=build /opt/myenv /opt/myenv

 ENV PATH="/opt/myenv/bin:$PATH"