diff --git a/setup.py b/setup.py index fedbc370f72..1dd8676e8eb 100644 --- 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, ]