From a9d87f34d1adc48c352d192acb556ac83d85c497 Mon Sep 17 00:00:00 2001
From: Artem Baranovskyi <artem.baranovsky1980@gmail.com>
Date: Wed, 8 Jan 2025 20:40:10 +0200
Subject: [PATCH] Image build is optimized.

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

diff --git a/flask_ml_api/Dockerfile b/flask_ml_api/Dockerfile
index 58fabee..1645827 100755
--- a/flask_ml_api/Dockerfile
+++ b/flask_ml_api/Dockerfile
@@ -3,9 +3,11 @@ 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
 RUN chmod +x /usr/local/bin/rustup-init.sh
+RUN /usr/local/bin/rustup-init.sh -y --default-toolchain none
 
-RUN /root/.cargo/bin/rustup --version
-RUN /root/.cargo/bin/cargo --version
+ENV PATH="$PATH:/root/.cargo/bin"
+
+RUN /root/.cargo/bin/rustup toolchain install stable --profile minimal --no-self-update
 
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential python3-venv && \
-- 
GitLab