File tree 2 files changed +3
-3
lines changed
2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -43,7 +43,7 @@ COPY . ./intel-extension-for-pytorch
43
43
RUN . ~/miniforge3/bin/activate && conda create -y -n compile_py310 python=3.10 && conda activate compile_py310 && \
44
44
cd intel-extension-for-pytorch/examples/cpu/llm && \
45
45
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 && \
47
47
unset CC && unset CXX
48
48
49
49
FROM base AS deploy
@@ -60,7 +60,7 @@ COPY --from=dev /root/intel-extension-for-pytorch/tools/get_libstdcpp_lib.sh ./l
60
60
RUN . ~/miniforge3/bin/activate && conda create -y -n py310 python=3.10 && conda activate py310 && \
61
61
cd /usr/lib/x86_64-linux-gnu/ && ln -s libtcmalloc.so.4 libtcmalloc.so && cd && \
62
62
cd ./llm && \
63
- bash tools/env_setup.sh 1 && \
63
+ bash tools/env_setup.sh 9 && \
64
64
python -m pip cache purge && \
65
65
mv ./oneCCL_release /opt/oneCCL && \
66
66
chown -R root:root /opt/oneCCL && \
Original file line number Diff line number Diff line change @@ -51,7 +51,7 @@ conda activate llm
51
51
52
52
# Setup the environment with the provided script
53
53
cd examples/cpu/llm
54
- bash ./tools/env_setup.sh 3
54
+ bash ./tools/env_setup.sh 11
55
55
56
56
# Activate environment variables
57
57
# set bash script argument to "inference" or "fine-tuning" for different usages
You can’t perform that action at this time.
0 commit comments