"cmake/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "2694b12725923d66d73b6f3f7ffa05a4db0cb10a"
Unverified Commit 46aeb086 authored by Wang, Yi's avatar Wang, Yi Committed by GitHub
Browse files

add xpu triton in dockerfile, or will show "Could not import Flash At… (#2702)



add xpu triton in dockerfile, or will show "Could not import Flash Attention enabled models: No module named 'triton'"
Signed-off-by: default avatarWang, Yi A <yi.a.wang@intel.com>
parent 98330df6
...@@ -93,6 +93,7 @@ ENV HF_HOME=/data \ ...@@ -93,6 +93,7 @@ ENV HF_HOME=/data \
WORKDIR /usr/src 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 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 # Install server
COPY proto proto COPY proto proto
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment