diff --git a/setup.py b/setup.py
index 3377b9ce5084c3f82ba1d8bf28903eb462ec44ae..2003818002252c022705bf37c4a79601d9ab30ad 100644
--- a/setup.py
+++ b/setup.py
@@ -16,8 +16,8 @@ setuptools.setup(
     install_requires=[
         "matplotlib>=3.3",
         "networkx>=2.5",
-        "torch==1.7.1",
-        "torchvision==0.8.2",
+        "torch>=1.5.1",
+        "torchvision>=0.6",
         "tqdm>=4.50",
         "pandas>=1.1",
         "jsonpickle>=1.5",