diff --git a/Dockerfile_intel b/Dockerfile_intel index 17570ee8..3996ef5b 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==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/test/xpu +RUN pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/test/xpu RUN pip install triton-xpu==3.2.0b1 --no-cache-dir