Skip to content

Commit b164702

Browse files
authored
install deepspeed from commit (#3663)
1 parent df995d4 commit b164702

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

examples/cpu/llm/Dockerfile

+2-2
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ COPY . ./intel-extension-for-pytorch
4343
RUN . ~/miniforge3/bin/activate && conda create -y -n compile_py310 python=3.10 && conda activate compile_py310 && \
4444
cd intel-extension-for-pytorch/examples/cpu/llm && \
4545
export CC=gcc && export CXX=g++ && \
46-
if [ -z ${COMPILE} ]; then bash tools/env_setup.sh 6; else bash tools/env_setup.sh 2; fi && \
46+
if [ -z ${COMPILE} ]; then bash tools/env_setup.sh 14; else bash tools/env_setup.sh 10; fi && \
4747
unset CC && unset CXX
4848

4949
FROM base AS deploy
@@ -60,7 +60,7 @@ COPY --from=dev /root/intel-extension-for-pytorch/tools/get_libstdcpp_lib.sh ./l
6060
RUN . ~/miniforge3/bin/activate && conda create -y -n py310 python=3.10 && conda activate py310 && \
6161
cd /usr/lib/x86_64-linux-gnu/ && ln -s libtcmalloc.so.4 libtcmalloc.so && cd && \
6262
cd ./llm && \
63-
bash tools/env_setup.sh 1 && \
63+
bash tools/env_setup.sh 9 && \
6464
python -m pip cache purge && \
6565
mv ./oneCCL_release /opt/oneCCL && \
6666
chown -R root:root /opt/oneCCL && \

examples/cpu/llm/README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ conda activate llm
5151

5252
# Setup the environment with the provided script
5353
cd examples/cpu/llm
54-
bash ./tools/env_setup.sh 3
54+
bash ./tools/env_setup.sh 11
5555

5656
# Activate environment variables
5757
# set bash script argument to "inference" or "fine-tuning" for different usages

0 commit comments

Comments
 (0)