diff --git a/Dockerfile_intel b/Dockerfile_intel index bacefd020..10ed914a7 100644 --- a/Dockerfile_intel +++ b/Dockerfile_intel @@ -98,7 +98,7 @@ ENV HF_HOME=/data \ WORKDIR /usr/src -RUN pip install torch==2.6.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/test/xpu +RUN pip install torch==2.6.0 --index-url https://download.pytorch.org/whl/test/xpu RUN pip install triton-xpu==3.2.0b1 --no-cache-dir