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',