From 80a9dc6ae4ad48fb5903c92d94eff377cc20fdc6 Mon Sep 17 00:00:00 2001
From: hailin <hailin@gdzx.xyz>
Date: Thu, 4 Sep 2025 17:34:52 +0800
Subject: [PATCH] .

---
 mm-zero3.sh | 5 +++++
 1 file changed, 5 insertions(+)

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