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

fix wrong cpu core num seen by docker (#10645)

Browse source
This commit is contained in:
Heyang Sun 2024-04-03 15:52:25 +08:00 committed by GitHub
parent 702e686901
commit 4f6df37805
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
docker/llm/finetune/qlora/cpu/docker/start-qlora-finetuning-on-cpu.sh
View file

@ -4,6 +4,7 @@ cd /ipex_llm
export USE_XETLA=OFF export USE_XETLA=OFF
export SYCL_PI_LEVEL_ZERO_USE_IMMEDIATE_COMMANDLISTS=1 export SYCL_PI_LEVEL_ZERO_USE_IMMEDIATE_COMMANDLISTS=1
source /opt/intel/oneapi/setvars.sh source /opt/intel/oneapi/setvars.sh
export CPU_CORES=$(nproc)
source ipex-llm-init -t source ipex-llm-init -t
if [ -d "./model" ]; if [ -d "./model" ];
@ -19,7 +20,6 @@ fi
if [ "$STANDALONE_DOCKER" = "TRUE" ] if [ "$STANDALONE_DOCKER" = "TRUE" ]
then then
export CONTAINER_IP=$(hostname -i) export CONTAINER_IP=$(hostname -i)
export CPU_CORES=$(nproc)
source /opt/intel/oneapi/setvars.sh source /opt/intel/oneapi/setvars.sh
export CCL_WORKER_COUNT=$WORKER_COUNT_DOCKER export CCL_WORKER_COUNT=$WORKER_COUNT_DOCKER
export CCL_WORKER_AFFINITY=auto export CCL_WORKER_AFFINITY=auto