From ef0ebcb770941f5fc866167306b11a1e2bc9ec07 Mon Sep 17 00:00:00 2001
From: hofee <lexhofee@gmail.com>
Date: Fri, 13 Sep 2024 17:29:11 +0800
Subject: [PATCH] add install requires

---
 setup.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/setup.py b/setup.py
index 0696f6c..22503ca 100644
--- 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',