diff --git a/setup.py b/setup.py
index fba55fc8..97a8612c 100644
--- a/setup.py
+++ b/setup.py
@@ -107,7 +107,8 @@ def get_extensions():
     return extensions
 
 
-install_requires = []
+install_requires = ["torch>=1.8.0"]
+extra_index_url = ["https://download.pytorch.org/whl/"]
 
 test_requires = [
     'pytest',
@@ -130,6 +131,7 @@ def get_extensions():
     keywords=['pytorch', 'scatter', 'segment', 'gather'],
     python_requires='>=3.8',
     install_requires=install_requires,
+    extra_index_url=extra_index_url,
     extras_require={
         'test': test_requires,
     },