Update pyproject.toml to try to only allow cpu versions of torch for now

ewanwm · Sep 20, 2024 · 19df3ac · 19df3ac
1 parent f7299ce
commit 19df3ac
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,5 +1,5 @@
 [build-system]
-requires = ["scikit-build-core", "pybind11", "torch"]
+requires = ["scikit-build-core", "pybind11", "torch*cpu"]
 build-backend = "scikit_build_core.build"
 
 [project]
@@ -12,4 +12,4 @@ authors = [
 ]
 
 [tool.scikit-build]
-cmake.args = ["-DNT_ENABLE_PYTHON=ON"]
+cmake.args = ["-DNT_ENABLE_PYTHON=ON"]