diff options
Diffstat (limited to 'setup.py')
-rw-r--r-- | setup.py | 7 |
1 files changed, 6 insertions, 1 deletions
@@ -76,7 +76,12 @@ setup( keywords=["ethos-u", "vela compiler", "tflite", "npu"], packages=find_namespace_packages(include=["ethosu.*"]), python_requires="~=3.6", # We support only 3.6+ - install_requires=["flatbuffers==1.11.0", "numpy>=1.16.6", "lxml>=4.5.1"], + install_requires=[ + "flatbuffers==1.11.0", + "numpy>=1.16.6", + "numpy>=1.16.6,<1.19.4 ; platform_system=='Windows'", + "lxml>=4.5.1", + ], entry_points={"console_scripts": ["vela = ethosu.vela.vela:main"]}, ext_modules=[mlw_module, tensor_allocator_module], setup_requires=["setuptools_scm"], |