From f7d47cb35ca99cb5dc2ce215a190ff318c33d658 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Tue, 18 Mar 2025 14:53:56 +0100 Subject: [PATCH] Fixing dockerfile ? --- Dockerfile_intel | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile_intel b/Dockerfile_intel index 3996ef5b..10ed914a 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.5.0 --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