ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Modify CPU finetune docker for bz2 error (#10919)

Browse source 
* Avoid bz2 error
* change to cpu torch
This commit is contained in:
Qiyuan Gong 2024-05-06 10:41:50 +08:00 committed by GitHub
parent 0e0bd309e2
commit 41ffe1526c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
docker/llm/finetune/qlora/cpu/docker/Dockerfile
View file

@ -20,7 +20,8 @@ RUN echo "deb [signed-by=/usr/share/keyrings/intel-oneapi-archive-keyring.gpg] h
RUN mkdir -p /ipex_llm/data && mkdir -p /ipex_llm/model && \
# Install python 3.11.1
apt-get update && apt-get install -y curl wget gpg gpg-agent software-properties-common git gcc g++ make libunwind8-dev zlib1g-dev libssl-dev libffi-dev && \
apt-get update && \
apt-get install -y curl wget gpg gpg-agent software-properties-common git gcc g++ make libunwind8-dev libbz2-dev zlib1g-dev libssl-dev libffi-dev && \
mkdir -p /opt/python && \
cd /opt/python && \
wget https://www.python.org/ftp/python/3.11.1/Python-3.11.1.tar.xz && \
@ -39,7 +40,8 @@ RUN mkdir -p /ipex_llm/data && mkdir -p /ipex_llm/model && \
rm -rf /var/lib/apt/lists/* && \
pip install --upgrade pip && \
export PIP_DEFAULT_TIMEOUT=100 && \
pip install --upgrade torch==2.1.0 && \
# install torch CPU version
pip install --upgrade torch==2.1.0 --index-url https://download.pytorch.org/whl/cpu && \
# install CPU ipex-llm
pip install --pre --upgrade ipex-llm[all] && \
# install ipex and oneccl