diff --git a/requirements.txt b/requirements.txt index 1cf243d912..882788d812 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,4 @@ -six -numpy +numpy >= 1.18.0 +onnx >= 1.2.3 protobuf +six diff --git a/setup.py b/setup.py index 34d77b699f..87180cff61 100644 --- a/setup.py +++ b/setup.py @@ -201,6 +201,9 @@ if bdist_wheel is not None : cmd_classes['bdist_wheel'] = bdist_wheel cmd_classes['build_ext'] = build_ext +with open('requirements.txt') as f: + install_requires = f.read().splitlines() + # Setup setup( name=package_name, @@ -222,10 +225,7 @@ setup( 'onnxruntime': data + examples + extra, }, py_modules=python_modules_list, - install_requires=[ - 'onnx>=1.2.3', - 'numpy>=1.18.0' - ], + install_requires=install_requires, entry_points= { 'console_scripts': [ 'onnxruntime_test = onnxruntime.tools.onnxruntime_test:main',