diff --git a/config.py b/config.py index 959b2b7..f237d51 100644 --- a/config.py +++ b/config.py @@ -1,4 +1,5 @@ import argparse +import sys import torch from multiprocessing import cpu_count @@ -33,10 +34,11 @@ class Config: @staticmethod def arg_parse() -> tuple: + exe = sys.executable or "python" parser = argparse.ArgumentParser() parser.add_argument("--port", type=int, default=7865, help="Listen port") parser.add_argument( - "--pycmd", type=str, default="python", help="Python command" + "--pycmd", type=str, default=exe, help="Python command" ) parser.add_argument("--colab", action="store_true", help="Launch in colab") parser.add_argument(