Update Dockerfile (#13081)
This commit is contained in:
parent
fa56212bb3
commit
db5edba786
1 changed files with 2 additions and 2 deletions
|
|
@ -53,7 +53,7 @@ RUN set -eux && \
|
|||
dpkg -i *.deb && \
|
||||
#
|
||||
# Install Intel PyTorch extension for LLM inference
|
||||
pip install --pre --upgrade ipex-llm[xpu_2.6]==2.3.0b20250407 --extra-index-url https://download.pytorch.org/whl/xpu && \
|
||||
pip install --pre --upgrade ipex-llm[xpu_2.6] --extra-index-url https://download.pytorch.org/whl/xpu && \
|
||||
#
|
||||
# Build torch-ccl
|
||||
mkdir -p /build && \
|
||||
|
|
@ -123,7 +123,7 @@ RUN set -eux && \
|
|||
wget https://bootstrap.pypa.io/get-pip.py -O get-pip.py && \
|
||||
python3 get-pip.py && rm get-pip.py && \
|
||||
pip install --upgrade requests argparse urllib3 && \
|
||||
pip install --pre --upgrade ipex-llm[xpu_2.6]==2.3.0b20250407 --extra-index-url https://download.pytorch.org/whl/xpu && \
|
||||
pip install --pre --upgrade ipex-llm[xpu_2.6] --extra-index-url https://download.pytorch.org/whl/xpu && \
|
||||
pip install transformers_stream_generator einops tiktoken && \
|
||||
pip install --upgrade colorama && \
|
||||
#
|
||||
|
|
|
|||
Loading…
Reference in a new issue