MLBEDSW-6979: Installing on aarch64 with Python 3.8 fails
- The issue is due to the numpy version needed when installing on
aarch64 with Python 3.8 and TensorFlow
- The fix is to use the python_version variable when specifing the
numpy version
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I6134b6dbccefc3be0b87feb17e3176b7f42641b3
diff --git a/setup.py b/setup.py
index b3011f8..3eb79b6 100644
--- a/setup.py
+++ b/setup.py
@@ -94,11 +94,12 @@
python_requires="~=3.7",
install_requires=[
"flatbuffers==2.0.7",
- "numpy<=1.21.3",
+ "numpy<=1.21.3; python_version<='3.7'",
+ "numpy; python_version>'3.7'",
"lxml>=4.5.1",
],
entry_points={"console_scripts": ["vela = ethosu.vela.vela:main"]},
ext_modules=[mlw_module],
cmdclass={"build_ext": BuildExtension}, # type: ignore[dict-item]
- setup_requires=["numpy<=1.21.3", "setuptools_scm"],
+ setup_requires=["numpy<=1.21.3; python_version<='3.7'", "numpy; python_version>'3.7'", "setuptools_scm<6.0"],
)