Use numpy 1.26.0

pytorch · Jun 19, 2024 · c387948 · c387948
1 parent 3f14196
commit c387948
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -18,7 +18,7 @@ pyyaml
 # We need to pin numpy version to the same as the torch testing environment
 # which still supports python 3.8
 numpy==1.21.2; python_version < '3.11'
-numpy==2.0.0; python_version >= '3.11'
+numpy==1.26.0; python_version >= '3.11'
 opencv-python
 submitit
 pynvml
diff --git a/torchbenchmark/models/Background_Matting/requirements.txt b/torchbenchmark/models/Background_Matting/requirements.txt
@@ -1,7 +1,7 @@
 # We need to pin numpy version to the same as the torch testing environment
 # which still supports python 3.8
 numpy==1.21.2; python_version < '3.11'
-numpy==2.0.0; python_version >= '3.11'
+numpy==1.26.0; python_version >= '3.11'
 opencv-python
 pandas
 Pillow

diff --git a/torchbenchmark/models/hf_Whisper/requirements.txt b/torchbenchmark/models/hf_Whisper/requirements.txt
@@ -2,4 +2,4 @@ numba
 # We need to pin numpy version to the same as the torch testing environment
 # which still supports python 3.8
 numpy==1.21.2; python_version < '3.11'
-numpy==2.0.0; python_version >= '3.11'
+numpy==1.26.0; python_version >= '3.11'
diff --git a/torchbenchmark/models/tacotron2/requirements.txt b/torchbenchmark/models/tacotron2/requirements.txt
@@ -1,7 +1,7 @@
 # We need to pin numpy version to the same as the torch testing environment
 # which still supports python 3.8
 numpy==1.21.2; python_version < '3.11'
-numpy==2.0.0; python_version >= '3.11'
+numpy==1.26.0; python_version >= '3.11'
 inflect
 scipy
 Unidecode

diff --git a/torchbenchmark/util/framework/detectron2/requirements.txt b/torchbenchmark/util/framework/detectron2/requirements.txt
@@ -3,4 +3,4 @@ omegaconf==2.3.0
 # We need to pin numpy version to the same as the torch testing environment
 # which still supports python 3.8
 numpy==1.21.2; python_version < '3.11'
-numpy==2.0.0; python_version >= '3.11'
+numpy==1.26.0; python_version >= '3.11'