From 95aee7bb4c5b0b7df7b452cd9fd1ed872fc553cd Mon Sep 17 00:00:00 2001
From: Artem Baranovskyi <artem.baranovsky1980@gmail.com>
Date: Mon, 6 Jan 2025 18:21:59 +0200
Subject: [PATCH] Image build is optimized.
---
.gitlab-ci.yml | 7 +------
flask_ml_api/Dockerfile | 13 ++-----------
2 files changed, 3 insertions(+), 17 deletions(-)
diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 9e32f8a..d907344 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -42,16 +42,12 @@ after_script:
build:
stage: build
image: docker:latest
-# image: docker:dind
services:
- name: docker:dind
- command: ["--host=fd://", "--tls=false", "--storage-driver=overlay2", "--mtu=1500", "--data-root=/mnt/volume"]
+ command: ["--host=tcp://0.0.0.0:2375", "--tls=false", "--storage-driver=overlay2", "--mtu=1500", "--data-root=/mnt/volume"]
script:
- docker buildx create --use
- # - docker build --cache-from=flask-backend-full --no-cache --progress=plain -t flask-backend-full -f flask_ml_api/Dockerfile .
- docker build --cache-from=22baar1mst/asyst-nlp-grader --no-cache --progress=plain -t 22baar1mst/asyst-nlp-grader:latest -f flask_ml_api/Dockerfile .
-# - docker build --no-cache --progress=plain -t 22baar1mst/asyst-nlp-grader:latest -f flask_ml_api/Dockerfile .
- - echo $DOCKER_REPO
- docker push 22baar1mst/asyst-nlp-grader:latest
only:
- master
@@ -59,7 +55,6 @@ build:
deploy:
stage: deploy
image: docker:latest
-# image: docker:dind
services:
- name: docker:dind
command: ["--tls=false", "--storage-driver=overlay2", "--mtu=1500", "--data-root=/mnt/volume"]
diff --git a/flask_ml_api/Dockerfile b/flask_ml_api/Dockerfile
index 8944451..bdc1fc0 100755
--- a/flask_ml_api/Dockerfile
+++ b/flask_ml_api/Dockerfile
@@ -1,4 +1,3 @@
-#FROM pytorch/pytorch:2.0.1-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 \
@@ -14,19 +13,11 @@ COPY . /app
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
+ rm -rf /root/.cache/pip
FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime
+
COPY --from=build /opt/myenv /opt/myenv
ENV PATH="/opt/myenv/bin:$PATH"
--
GitLab