openvino/tests/requirements_tensorflow at 3226244667832091bf9e6bc14767faadb122adab · openvinotoolkit/openvino · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
# test ovc with NumPy 2.x on Ubuntu 24 with default Python 3.12
# test against NumPy 1.x with older Python versions
# tensorflow-intel 2.18.0 depends on numpy<2.1.0 and >=1.26.0
# tensorflow 2.16.2 depends on numpy<2.0.0 and >=1.26.0; python_version >= "3.12"
numpy==1.26.4; python_version < "3.12" or (platform_system == "Darwin" and platform_machine == "x86_64" and python_version < "3.13")
numpy==2.2.6; python_version >= "3.13" or (python_version >= "3.12" and (platform_system != "Darwin" or platform_machine != "x86_64"))
pytest==7.2.0
pytest-xdist[psutil]==3.6.1
pytest-html==4.2.0
transformers==4.45.1
# install exact keras version since tensorflow depends and has no upper bound for it
keras==3.13.0; python_version >= "3.13"
keras==3.12.0; python_version < "3.13"
tensorflow==2.20.0; python_version >= "3.13" or (python_version >= "3.11" and (platform_system != "Darwin" or platform_machine != "x86_64"))
tensorflow==2.16.2; python_version < "3.13" and (python_version < "3.11" or (platform_system == "Darwin" and platform_machine == "x86_64"))
# install explicit version of wrapt to avoid "this __dict__ descriptor does not support '_DictWrapper' objects" error from TensorFlow 2.18
wrapt==1.15.0; python_version >= "3.12"
# tensorflow-text is not available for both Windows and ARM platforms
#tensorflow-text==2.19.0; python_version < "3.13" and platform_system == "Linux" and platform_machine == "x86_64" # ticket 179327
tensorflow-hub==0.16.1
jax==0.5.0; python_version >= "3.11" and (platform_system != "Darwin" or platform_machine != "x86_64")
jax==0.4.33; python_version < "3.11" or (platform_system == "Darwin" and platform_machine == "x86_64")
defusedxml==0.7.1