add install requires

2024-09-13 17:29:11 +08:00 · 2024-09-13 17:29:11 +08:00 · ef0ebcb770
commit ef0ebcb770
parent 7546b34515
1 changed files with 9 additions and 0 deletions
--- a/setup.py
+++ b/setup.py
@ -9,6 +9,15 @@ setup(
                        "ui/client/static/css/*","ui/client/static/fonts/*",
                        "ui/client/static/img/*","ui/client/static/js/*"],
    },
+    install_requires=[                 
+        'torch',
+        'pyyaml',
+        'psutil',
+        'GPUtil',
+        'flask',
+        'flask_cors', 
+        'tensorboard',
+    ],
    entry_points={
        'console_scripts': [
            'pytorch-boot = PytorchBoot.boot:main',