From 46aeb0860dae0c5a1e5990dff50f8d381fddce61 Mon Sep 17 00:00:00 2001 From: "Wang, Yi" Date: Wed, 30 Oct 2024 21:18:50 +0800 Subject: [PATCH] =?UTF-8?q?add=20xpu=20triton=20in=20dockerfile,=20or=20wi?= =?UTF-8?q?ll=20show=20"Could=20not=20import=20Flash=20At=E2=80=A6=20(#270?= =?UTF-8?q?2)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit add xpu triton in dockerfile, or will show "Could not import Flash Attention enabled models: No module named 'triton'" Signed-off-by: Wang, Yi A --- Dockerfile_intel | 1 + 1 file changed, 1 insertion(+) diff --git a/Dockerfile_intel b/Dockerfile_intel index 96f24248..f9b1cd13 100644 --- a/Dockerfile_intel +++ b/Dockerfile_intel @@ -93,6 +93,7 @@ ENV HF_HOME=/data \ WORKDIR /usr/src RUN pip install torch==2.3.1+cxx11.abi torchvision==0.18.1+cxx11.abi torchaudio==2.3.1+cxx11.abi intel-extension-for-pytorch==2.3.110+xpu oneccl_bind_pt==2.3.100+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ --no-cache-dir +RUN pip install triton-xpu==3.0.0b2 --no-cache-dir # Install server COPY proto proto