.

2025-09-04 17:34:52 +08:00 · 2025-09-04 17:34:52 +08:00 · 80a9dc6ae4
parent 61b910c379
commit 80a9dc6ae4
1 changed files with 5 additions and 0 deletions
--- a/mm-zero3.sh
+++ b/mm-zero3.sh
@ -9,6 +9,11 @@ set -a
 . ./.deepspeed_env
 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 继承到各节点）
 unset DS_BUILD_OPS DS_SKIP_CUDA_BUILD PYTHONNOUSERSITE