[Fix] Update dlc runner python env (open-compass#1604)

bittersweet1999 · Oct 14, 2024 · f390697 · f390697
1 parent 4fde410
commit f390697
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/opencompass/runners/dlc.py b/opencompass/runners/dlc.py
@@ -182,7 +182,11 @@ def _launch(self, cfg: ConfigDict, random_sleep: Optional[bool] = None):
                               cfg_path=param_file,
                               template=tmpl)
             cmd = get_cmd()
-
+            # Use specified python env instead of sys.executable
+            if self.aliyun_cfg['python_env_path']:
+                cmd = cmd.replace(
+                    sys.executable,
+                    f'{self.aliyun_cfg["python_env_path"]}/bin/python')
             logger = get_logger()
             logger.debug(f'Running command: {cmd}')