From dfaf76659f8c402fc1c63db098587663bb9d55b3 Mon Sep 17 00:00:00 2001
From: Dan Saunders <danjsaund@gmail.com>
Date: Fri, 26 Sep 2025 19:53:51 -0400
Subject: [PATCH] pip install --system flag

---
 docker/Dockerfile         |  6 +++---
 docker/Dockerfile-uv-base | 10 +++++-----
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 6c939f219..b0f7d8a88 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -29,13 +29,13 @@ RUN git fetch origin "$GIT_REF" && git checkout "$GIT_SHA"
 
 # If AXOLOTL_EXTRAS is set, append it in brackets
 RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \
-        uv pip install --no-build-isolation -e .[ring-flash-attn,optimizers,ray,$AXOLOTL_EXTRAS] $AXOLOTL_ARGS; \
+        uv pip install --system --no-build-isolation -e .[ring-flash-attn,optimizers,ray,$AXOLOTL_EXTRAS] $AXOLOTL_ARGS; \
     else \
-        uv pip install --no-build-isolation -e .[ring-flash-attn,optimizers,ray] $AXOLOTL_ARGS; \
+        uv pip install --system --no-build-isolation -e .[ring-flash-attn,optimizers,ray] $AXOLOTL_ARGS; \
     fi && \
     python scripts/unsloth_install.py | sh && \
     python scripts/cutcrossentropy_install.py | sh && \
-    uv pip install pytest
+    uv pip install --system pytest
 
 # fix so that git fetch/pull from remote works with shallow clone
 RUN git config remote.origin.fetch "+refs/heads/*:refs/remotes/origin/*" && \
diff --git a/docker/Dockerfile-uv-base b/docker/Dockerfile-uv-base
index 4b08e55f8..5c25df120 100644
--- a/docker/Dockerfile-uv-base
+++ b/docker/Dockerfile-uv-base
@@ -29,8 +29,8 @@ RUN uv venv --no-project --relocatable axolotl-venv
 
 ENV PATH="/workspace/axolotl-venv/bin:${PATH}"
 
-RUN uv pip install packaging setuptools wheel psutil \
-    && uv pip install torch==${PYTORCH_VERSION} \
-    && uv pip install --no-build-isolation "causal_conv1d @ git+https://github.com/Dao-AILab/causal-conv1d.git@main" \
-    && uv pip install "mamba_ssm @ git+https://github.com/state-spaces/mamba.git@main" \
-    && uv pip install awscli pydantic
+RUN uv pip install --system packaging setuptools wheel psutil \
+    && uv pip install --system torch==${PYTORCH_VERSION} \
+    && uv pip install --system --no-build-isolation "causal_conv1d @ git+https://github.com/Dao-AILab/causal-conv1d.git@main" \
+    && uv pip install --system "mamba_ssm @ git+https://github.com/state-spaces/mamba.git@main" \
+    && uv pip install --system awscli pydantic