From 842a2860af3bb736f15b522fb0c4d4f0aad85ce5 Mon Sep 17 00:00:00 2001 From: Peter Dave Hello Date: Fri, 19 May 2023 22:41:14 +0800 Subject: [PATCH] Optimize pip install in Dockerfile to minimize image size --- docker/Dockerfile | 2 +- docker/cuda.Dockerfile | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index fb4f35d..4885701 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -9,7 +9,7 @@ RUN apt-get update -qq \ && apt-get clean \ && rm -rf /var/lib/apt -RUN python -mvenv venv && ./venv/bin/pip install --upgrade pip +RUN python -mvenv venv && ./venv/bin/pip install --no-cache-dir --upgrade pip COPY . . diff --git a/docker/cuda.Dockerfile b/docker/cuda.Dockerfile index 232f8a6..1b29547 100644 --- a/docker/cuda.Dockerfile +++ b/docker/cuda.Dockerfile @@ -13,17 +13,17 @@ RUN apt-get update -qq \ && apt-get clean \ && rm -rf /var/lib/apt -RUN pip3 install --upgrade pip && apt-get remove python3-pip --assume-yes +RUN pip3 install --no-cache-dir --upgrade pip && apt-get remove python3-pip --assume-yes RUN ln -s /usr/bin/python3 /usr/bin/python -RUN pip3 install torch==1.12.0+cu116 -f https://download.pytorch.org/whl/torch_stable.html +RUN pip3 install --no-cache-dir torch==1.12.0+cu116 -f https://download.pytorch.org/whl/torch_stable.html COPY . . RUN if [ "$with_models" = "true" ]; then \ # install only the dependencies first - pip3 install -e .; \ + pip3 install --no-cache-dir -e .; \ # initialize the language models if [ ! -z "$models" ]; then \ ./scripts/install_models.py --load_only_lang_codes "$models"; \