Adjusting Requirements

ericspod · ericspod · commit 5c0b5f6e3074 · 2024-08-16T17:36:04.000+01:00
diff --git a/environment-dev.yml b/environment-dev.yml
@@ -5,10 +5,10 @@ channels:
   - nvidia
   - conda-forge
 dependencies:
-  - numpy>=1.20
+  - numpy>=1.25,<2.0
   - pytorch>=1.9
   - torchvision
-  - pytorch-cuda=11.6
+  - pytorch-cuda>=11.6
   - pip
   - pip:
     - -r requirements-dev.txt
diff --git a/requirements.txt b/requirements.txt
@@ -1,2 +1,2 @@
 torch>=1.9
-numpy>=1.25
+numpy>=1.25,<2.0
diff --git a/setup.cfg b/setup.cfg
@@ -42,7 +42,7 @@ setup_requires =
     ninja
 install_requires =
     torch>=1.9
-    numpy>=1.20
+    numpy>=1.25,<2.0
 
 [options.extras_require]
 all =
