update

microsoft · Jun 26, 2024 · 2278aa4 · 2278aa4
1 parent 3c0b407
commit 2278aa4
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/tools/ci_build/github/linux/docker/scripts/lort/requirements.txt b/tools/ci_build/github/linux/docker/scripts/lort/requirements.txt
@@ -8,7 +8,8 @@ onnx==1.16.1
 astunparse
 expecttest!=0.2.0
 hypothesis
-numpy
+numpy==1.21.6 ; python_version < '3.9'
+numpy==2.0.0 ; python_version >= '3.9'
 psutil
 pyyaml
 requests

diff --git a/...ld/github/linux/docker/scripts/training/ortmodule/stage1/torch_eager_cpu/requirements.txt b/...ld/github/linux/docker/scripts/training/ortmodule/stage1/torch_eager_cpu/requirements.txt
@@ -5,6 +5,7 @@ setuptools>=68.2.2
 cerberus
 h5py
 scikit-learn
-numpy
+numpy==1.21.6 ; python_version < '3.9'
+numpy==2.0.0 ; python_version >= '3.9'
 pandas
 parameterized
diff --git a/tools/ci_build/github/windows/eager/requirements.txt b/tools/ci_build/github/windows/eager/requirements.txt
@@ -1,6 +1,7 @@
 setuptools
 wheel
-numpy
+numpy==1.21.6 ; python_version < '3.9'
+numpy==2.0.0 ; python_version >= '3.9'
 typing_extensions
 torch==1.13.1
 parameterized