Fixing dockerfile ?

This commit is contained in:
Nicolas Patry 2025-03-18 14:53:56 +01:00
parent b77281ac22
commit f7d47cb35c
No known key found for this signature in database
GPG Key ID: 4242CEF24CB6DBF9

View File

@ -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