Update Dockerfile (#13081)

This commit is contained in:
Shaojun Liu 2025-04-16 09:18:46 +08:00 committed by GitHub
parent fa56212bb3
commit db5edba786
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -53,7 +53,7 @@ RUN set -eux && \
dpkg -i *.deb && \ dpkg -i *.deb && \
# #
# Install Intel PyTorch extension for LLM inference # 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 # Build torch-ccl
mkdir -p /build && \ mkdir -p /build && \
@ -123,7 +123,7 @@ RUN set -eux && \
wget https://bootstrap.pypa.io/get-pip.py -O get-pip.py && \ wget https://bootstrap.pypa.io/get-pip.py -O get-pip.py && \
python3 get-pip.py && rm get-pip.py && \ python3 get-pip.py && rm get-pip.py && \
pip install --upgrade requests argparse urllib3 && \ 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 transformers_stream_generator einops tiktoken && \
pip install --upgrade colorama && \ pip install --upgrade colorama && \
# #