Fix build error of bigdl-llm-cpu (#10228)
This commit is contained in:
parent
19260492c7
commit
e08c74f1d1
1 changed files with 0 additions and 3 deletions
|
|
@ -65,11 +65,8 @@ RUN cp /root/miniconda3/condabin/conda /usr/bin && \
|
|||
pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cpu && \
|
||||
pip install intel-extension-for-pytorch==2.2.0 && \
|
||||
pip install oneccl_bind_pt==2.2.0 --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/cpu/us/ && \
|
||||
cd ./intel-extension-for-pytorch && \
|
||||
pip install -r requirements.txt && \
|
||||
pip install transformers==4.36.2 && \
|
||||
pip install transformers_stream_generator && \
|
||||
cd /llm && \
|
||||
echo "conda deactivate" >> /root/.bashrc
|
||||
|
||||
ENTRYPOINT ["/bin/bash"]
|
||||
|
|
|
|||
Loading…
Reference in a new issue