chore(no_model): Improve build time caching

jim60105 · Aug 30, 2023 · a1f3ffe · a1f3ffe
1 parent 6e1e1bd
commit a1f3ffe
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Dockerfile.no_model b/Dockerfile.no_model
@@ -6,17 +6,17 @@ FROM python:3.10-slim as dependencies
 # Setup venv
 RUN python3 -m venv /venv
 ARG PATH="/venv/bin:$PATH"
-RUN python3 -m pip install --upgrade pip setuptools
+RUN --mount=type=cache,target=/root/.cache/pip pip install --upgrade pip setuptools
 # Install requirements
-RUN python3 -m pip install torch torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
+RUN --mount=type=cache,target=/root/.cache/pip pip install torch torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
 
 # Add git
 ARG DEBIAN_FRONTEND=noninteractive
 RUN apt-get update && apt-get install -y --no-install-recommends git
 
 # Install whisperX
 COPY ./whisperX /code
-RUN python3 -m pip install /code
+RUN --mount=type=cache,target=/root/.cache/pip pip install /code
 
 
 FROM python:3.10-slim