Commit 3a5e04df authored by Artem Baranovskyi's avatar Artem Baranovskyi
Browse files

Image build is optimized.

parent 95aee7bb
No related merge requests found
Pipeline #10741 failed with stages
in 4 minutes and 42 seconds
Showing with 5 additions and 1 deletion
+5 -1
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
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment