Image build is optimized.

flask_ml_api/Dockerfile

@@ -3,11 +3,9 @@ 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 /usr/local/bin/rustup toolchain install stable --profile minimal --no-self-update
 
-RUN /usr/local/bin/rustup --version
-RUN /usr/local/bin/cargo --version
+RUN /root/.cargo/bin/rustup --version
+RUN /root/.cargo/bin/cargo --version
 
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential python3-venv && \