diff --git a/opencompass/runners/local.py b/opencompass/runners/local.py index 9a335fe7..35f3928a 100644 --- a/opencompass/runners/local.py +++ b/opencompass/runners/local.py @@ -99,7 +99,7 @@ class LocalRunner(BaseRunner): tmpl = get_command_template(all_gpu_ids[:num_gpus]) cmd = task.get_command(cfg_path=param_file, template=tmpl) # run in subprocess if starts with torchrun etc. - if cmd.startswith('python'): + if 'python3 ' in cmd or 'python ' in cmd: task.run() else: subprocess.run(cmd, shell=True, text=True)