Skip to content

Commit 2119970

Browse files
committed
fix onnxruntime segment fault with tf 1.10
1 parent 15ac39c commit 2119970

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

ci_build/azure_pipelines/templates/setup.yml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,14 @@ steps:
55
set -ex
66
pip install pytest pytest-cov pytest-runner graphviz requests pyyaml pillow pandas
77
pip install $(CI_PIP_TF_NAME) $(CI_PIP_ONNX_NAME) $(CI_PIP_ONNX_BACKEND_NAME)
8+
9+
# TF 1.10 requires numpy <=1.14.5 and >=1.13.3, but onnxruntime 0.2.1 does not work with numpy <= 1.14.5
10+
# Upgrade numpy only within constraints from other packages if any.
11+
if [[ $CI_TF_VERSION == 1.10* ]] && [[ $CI_ONNX_BACKEND == "onnxruntime" ]] ;
12+
then
13+
pip install $(CI_PIP_ONNX_NAME) $(CI_PIP_ONNX_BACKEND_NAME) numpy --no-deps -U
14+
fi
15+
816
python setup.py install
917
pip freeze --all
1018
displayName: 'Setup Environment'

0 commit comments

Comments
 (0)