Image build is optimized.

3a5e04df · Artem Baranovskyi · 95aee7bb · 3a5e04df
Commit 3a5e04df authored 5 months ago by Artem Baranovskyi
Hide whitespace changes
Inline Side-by-side

Showing

with 5 additions and 1 deletion
+5 -1
--- a/flask_ml_api/Dockerfile
+++ b/flask_ml_api/Dockerfile
 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 rustup-init.sh
+RUN sh rustup-init.sh -y --default-toolchain none
+RUN ~/.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 && \
    apt-get clean && \
@@ -15,7 +20,6 @@ RUN pip install --no-cache-dir Flask matplotlib scikit-learn transformers pandas
    pip install --no-cache-dir --upgrade setuptools wheel && \
    rm -rf /root/.cache/pip

-
 FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime

 COPY --from=build /opt/myenv /opt/myenv