diff --git a/pyproject.toml b/pyproject.toml
new file mode 100644
index 0000000..b61abcd
--- /dev/null
+++ b/pyproject.toml
@@ -0,0 +1,3 @@
+[build-system]
+requires = ["torch>=2.1.0"]
+build-backend = "setuptools.build_meta"
diff --git a/setup.py b/setup.py
index 2f4a427..916f760 100644
--- a/setup.py
+++ b/setup.py
@@ -78,6 +78,9 @@ if __name__ == '__main__':
         name='deep_gemm',
         version='2.0.0' + revision,
         packages=find_packages('.'),
+        install_requires=[
+            'torch>=2.1.0',
+        ],
         package_data={
             'deep_gemm': [
                 'include/deep_gemm/**/*',