diff --git a/tests/tensor_parallel/test_tensor_parallel.py b/tests/tensor_parallel/test_tensor_parallel.py index 6174a613295..6efd01c6cfd 100644 --- a/tests/tensor_parallel/test_tensor_parallel.py +++ b/tests/tensor_parallel/test_tensor_parallel.py @@ -77,7 +77,7 @@ class TestTensorParallel(TestCasePlus): f"torchrun --nproc_per_node {self.nproc_per_node} --master_port {get_torch_dist_unique_port()} {tmp.name}" ).split() else: - cmd = ["python", tmp.name] + cmd = ["python3", tmp.name] # Note that the subprocess will be waited for here, and raise an error if not successful try: