diff --git a/mm-zero3.sh b/mm-zero3.sh
index 39ab346..4e489c1 100755
--- 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