Remove numpy<2.0 windows pin (#8594)

pytorch · Aug 15, 2024 · edb1c33 · edb1c33
1 parent 831879a
commit edb1c33
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -91,8 +91,7 @@ def get_dist(pkgname):
         pytorch_dep += "==" + os.getenv("PYTORCH_VERSION")
 
     requirements = [
-        # TODO: Remove <2 constraint! https://github.com/pytorch/vision/issues/8531
-        "numpy<2" if sys.platform == "win32" else "numpy",
+        "numpy",
         pytorch_dep,
     ]