Skip to content

Commit

Permalink
Modify CPU finetune docker for bz2 error (intel-analytics#10919)
Browse files Browse the repository at this point in the history
* Avoid bz2 error
* change to cpu torch
  • Loading branch information
qiyuangong authored May 6, 2024
1 parent 0e0bd30 commit 41ffe15
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions docker/llm/finetune/qlora/cpu/docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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 && \
Expand All @@ -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
Expand Down

0 comments on commit 41ffe15

Please sign in to comment.