Triton fix (#2995)

fix triton to 3.1.0 to fix ipex import issue Signed-off-by: Wang, Yi A <[email protected]>
huggingface · Feb 6, 2025 · 36223f8 · 36223f8
1 parent 0ef8c8a
commit 36223f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile_intel b/Dockerfile_intel
@@ -190,7 +190,7 @@ RUN pip install https://download.pytorch.org/whl/nightly/cpu/torch-2.5.0.dev2024
 RUN pip install https://download.pytorch.org/whl/nightly/cpu/torchvision-0.20.0.dev20240815%2Bcpu-cp311-cp311-linux_x86_64.whl
 RUN pip install https://download.pytorch.org/whl/nightly/cpu/torchaudio-2.4.0.dev20240815%2Bcpu-cp311-cp311-linux_x86_64.whl
 
-RUN pip install triton py-libnuma
+RUN pip install triton==3.1.0 py-libnuma
 
 WORKDIR /usr/src