From 940e64b792be04ec984e8a6293d3540ec604502a Mon Sep 17 00:00:00 2001
From: Artem Baranovskyi <artem.baranovsky1980@gmail.com>
Date: Mon, 6 Jan 2025 17:27:10 +0200
Subject: [PATCH] Image build is optimized.

---
 flask_ml_api/Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/flask_ml_api/Dockerfile b/flask_ml_api/Dockerfile
index c6c6c16..0c49c58 100755
--- a/flask_ml_api/Dockerfile
+++ b/flask_ml_api/Dockerfile
@@ -1,5 +1,5 @@
 #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"
-- 
GitLab