diff --git a/pyproject.toml b/pyproject.toml
new file mode 100644
index 0000000000..55b04d0110
--- /dev/null
+++ b/pyproject.toml
@@ -0,0 +1,2 @@
+[build-system]
+requires = ["setuptools", "torch"]
\ No newline at end of file
diff --git a/setup.py b/setup.py
index b51307a625..0e2b7424fa 100644
--- a/setup.py
+++ b/setup.py
@@ -193,6 +193,7 @@ def get_model_zoo_configs() -> List[str]:
         # NOTE: When adding new dependencies, if it is required at import time (in addition
         # to runtime), it probably needs to appear in docs/requirements.txt, or as a mock
         # in docs/conf.py
+        "torch>=2.3.1"
     ],
     extras_require={
         # optional dependencies, required by some features