diff --git a/setup.py b/setup.py
index 6f3a2c7..6759c0c 100644
--- a/setup.py
+++ b/setup.py
@@ -8,5 +8,5 @@ setup(
     # author_email='author@gmail.com',
     # description='Description of my package',
     packages=['priorConditionedAnnealing'],
-    # install_requires=['numpy >= 1.11.1', 'matplotlib >= 1.5.1'],
+    install_requires=['numpy', 'pytorch', 'perlin_noise', 'colorednoise', 'stable_baselines3'],
 )