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