Image build is optimized.

.gitlab-ci.yml

@@ -50,7 +50,9 @@ build:
   image: docker:20.10.16
   services:
     - name: docker:20.10.16-dind
-      command: ["--host=tcp://0.0.0.0:2375", "--tls=false", "--storage-driver=overlay2", "--mtu=1500", "--data-root=/mnt/volume"]
+      command: ["--host=tcp://docker:2375", "--tls=false", "--storage-driver=overlay2", "--mtu=1500", "--data-root=/mnt/volume"]
+      volumes:
+        - /var/run/docker.sock:/var/run/docker.sock
   script:
     - docker info
     - docker buildx create --use
@@ -69,4 +71,4 @@ deploy:
     - docker pull 22baar1mst/asyst-nlp-grader:latest
     - docker run -d 22baar1mst/asyst-nlp-grader:latest
   only:
-    - master
+    - master
\ No newline at end of file

flask_ml_api/Dockerfile

-FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime as build
+=FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime as build
 
 RUN apt-get update && apt-get install -y curl
 RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs -o /usr/local/bin/rustup-init.sh