From 58a277da49393c64252b72a0bb31f76a273a54ed Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Thu, 16 Jan 2025 15:01:15 +0100 Subject: [PATCH] Install system ? --- Dockerfile_amd | 2 +- Dockerfile_intel | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/Dockerfile_amd b/Dockerfile_amd index 3098ee16..7981724f 100644 --- a/Dockerfile_amd +++ b/Dockerfile_amd @@ -104,7 +104,7 @@ RUN case ${TARGETPLATFORM} in \ /opt/conda/bin/conda clean -ya # Install flash-attention, torch dependencies -RUN python3 -m pip install --upgrade pip uv && uv pip install numpy einops ninja joblib msgpack cmake --no-cache-dir && rm -rf /var/lib/apt/lists/* +RUN python3 -m pip install --upgrade pip uv && pip install numpy einops ninja joblib msgpack cmake --no-cache-dir && rm -rf /var/lib/apt/lists/* RUN conda install mkl=2021 ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/rocm/lib/:/opt/conda/lib/python3.11/site-packages/torch/lib:/opt/conda/lib/ diff --git a/Dockerfile_intel b/Dockerfile_intel index a0767cb0..ece227e5 100644 --- a/Dockerfile_intel +++ b/Dockerfile_intel @@ -110,8 +110,8 @@ COPY server server COPY server/Makefile server/Makefile RUN cd server && \ make gen-server && \ - uv pip install -U pip uv && \ - uv pip install ".[accelerate, compressed-tensors, peft, outlines]" --no-cache-dir + pip install -U pip uv && \ + uv pip install ".[accelerate, compressed-tensors, peft, outlines]" --no-cache-dir --system ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/intel/oneapi/pti/0.9/lib:/opt/conda/lib ENV CCL_ZE_IPC_EXCHANGE=sockets