Pin numpy dep to numpy<2 for windows (#8530)

pytorch · Jul 17, 2024 · 0bf8d14 · 0bf8d14
1 parent 33e47d8
commit 0bf8d14
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -58,7 +58,8 @@ def write_version_file():
     pytorch_dep += "==" + os.getenv("PYTORCH_VERSION")
 
 requirements = [
-    "numpy",
+    # TODO: Remove <2 constraint! https://github.com/pytorch/vision/issues/8531
+    "numpy<2" if sys.platform == "win32" else "numpy",
     pytorch_dep,
 ]