hailin
/
jd_train
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

This commit is contained in:
hailin 2025-09-04 17:34:52 +08:00
parent 61b910c379
commit 80a9dc6ae4
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mm-zero3.sh
View File

@ -9,6 +9,11 @@ set -a
. ./.deepspeed_env . ./.deepspeed_env
set +a set +a
unset PYTHONNOUSERSITE
USER_SITE=$(python3 -c 'import site;print(site.getusersitepackages())')
export PATH="$HOME/.local/bin:$PATH"
export PYTHONPATH="$USER_SITE:/home/test/jd_train:${PYTHONPATH:-}"
# 统一环境（会被 deepspeed 的 ssh 继承到各节点） # 统一环境（会被 deepspeed 的 ssh 继承到各节点）
unset DS_BUILD_OPS DS_SKIP_CUDA_BUILD PYTHONNOUSERSITE unset DS_BUILD_OPS DS_SKIP_CUDA_BUILD PYTHONNOUSERSITE