chore(setup): add torch to build requirements (#682)

chore(setup): add torch to build requirements
Megvii-BaseDetection · Nov 10, 2021 · 2c2dd13 · 2c2dd13
1 parent f778fdb
commit 2c2dd13
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -0,0 +1,7 @@
+[build-system]
+requires = [
+    "setuptools>=42",
+    "torch>=1.7",
+    "wheel",
+]
+build-backend = "setuptools.build_meta"