From a84e86a5b3942a6c4fe7c30219a4790a2e85d712 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Tue, 18 Mar 2025 10:43:02 +0100
Subject: [PATCH] Intel docker.

---
 Dockerfile_intel | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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