mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-19 22:02:06 +00:00
Intel docker. (#3121)
* Intel docker. * torchaudio ? * Fixing dockerfile ?
This commit is contained in:
parent
83fe45c15e
commit
67ce543e04
@ -98,7 +98,7 @@ ENV HF_HOME=/data \
|
|||||||
|
|
||||||
|
|
||||||
WORKDIR /usr/src
|
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
|
RUN pip install triton-xpu==3.2.0b1 --no-cache-dir
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user