[CLI] Add requirements and improve ease of use (#381)

This PR sets the requirements of the cli installation, and we improve
ease of use:

- flagscale pull just on host machine, so we moved some imports into
concrete functions

- To avoid installation error on the CPU without GPU, we fix the
exception

Co-authored-by: caozhou <caozhou@baai.ac.cn>

Author: Caozhou1995

flagscale/cli.py

@@ -4,7 +4,6 @@
 
 import click
 
-from run import main as run_main
 
 VERSION = "0.6.0"
 
@@ -27,6 +26,7 @@ def train(yaml_path):
     """
     Train model from yaml.
     """
+    from run import main as run_main
     click.echo(f"Start training from the yaml {yaml_path}...")
     yaml_path = os.path.abspath(yaml_path)
     config_path = os.path.dirname(yaml_path)
@@ -49,6 +49,7 @@ def serve(model_name, yaml_path=None):
     """
     Serve model from yaml.
     """
+    from run import main as run_main
     if yaml_path:
         if os.path.isabs(yaml_path):
             yaml_path = yaml_path
@@ -112,6 +113,7 @@ def serve(model_name, yaml_path=None):
     help="The path to save ckpt",
 )
 def pull(image_name, ckpt_name, ckpt_path):
+    "Docker pull image and git clone ckpt."
     # If ckpt_path is not provided, use the default download directory
     if ckpt_path is None:
         ckpt_path = os.path.join(os.getcwd(), "model_download")

setup.py

@@ -12,7 +12,7 @@ def is_nvidia_chip():
         )
         if result.returncode == 0:
             return True
-    except FileNotFoundError:
+    except Exception as e:
         pass
     return False
 
@@ -64,7 +64,14 @@ def run(self):
         "flag_scale.flagscale": ["**/*"],
         "flag_scale.examples": ["**/*"],
     },
-    install_requires=["click", "cryptography"],
+    install_requires=[
+        "click",
+        "cryptography",
+        "setuptools>=75.1.0",
+        "packaging>=24.1",
+        "importlib_metadata>=8.5.0"
+    ],
+
     entry_points={
         "console_scripts": [
             "flagscale=flag_scale.flagscale.cli:flagscale",