Test ORT 1.8

onnx · Jun 3, 2021 · 6cc9ada · 6cc9ada
1 parent 383e431
commit 6cc9ada
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 6 deletions.
diff --git a/.azure-pipelines/linux-CI-keras-applications-nightly.yml b/.azure-pipelines/linux-CI-keras-applications-nightly.yml
@@ -42,7 +42,7 @@ jobs:
         INSTALL_KERAS: pip install keras==2.3.1
         UNINSTALL_KERAS:
         INSTALL_TENSORFLOW: pip install tensorflow==1.15.0
-        INSTALL_ORT: pip install -i https://test.pypi.org/simple/ ort-nightly
+        INSTALL_ORT: pip install -i https://test.pypi.org/simple/ ort-nightly==1.8.0.dev202106023
         INSTALL_KERAS_RESNET: pip install keras-resnet
         INSTALL_TRANSFORMERS:
         NIGHTLY_BUILD_TEST: python run_all.py --exclude "test_keras_applications_v2.py"
@@ -64,7 +64,7 @@ jobs:
         INSTALL_KERAS:
         UNINSTALL_KERAS: pip uninstall keras -y
         INSTALL_TENSORFLOW: pip install tensorflow==2.2.0
-        INSTALL_ORT: pip install -i https://test.pypi.org/simple/ ort-nightly
+        INSTALL_ORT: pip install -i https://test.pypi.org/simple/ ort-nightly==1.8.0.dev202106023
         INSTALL_KERAS_RESNET: pip install keras-resnet
         INSTALL_TRANSFORMERS: pip install transformers==3.4.0
         NIGHTLY_BUILD_TEST: python run_all_v2.py
@@ -91,6 +91,7 @@ jobs:
       pip install git+https://github.com/microsoft/onnxconverter-common
       pip install -r requirements.txt
       pip install -r requirements-dev.txt
+      pip install flatbuffers
       $(INSTALL_ORT)
       pip install opencv-python
       pip install tqdm

diff --git a/.azure-pipelines/linux-conda-CI-tf-keras.yml b/.azure-pipelines/linux-conda-CI-tf-keras.yml
@@ -42,7 +42,7 @@ jobs:
         python.version: '3.8'
         ONNX_PATH: onnx==1.8.0
         TENSORFLOW_PATH: tensorflow-cpu==2.3.0
-        INSTALL_ORT: pip install onnxruntime==1.7.0
+        INSTALL_ORT: pip install -i https://test.pypi.org/simple/ ort-nightly==1.8.0.dev202106023
 
     maxParallel: 3
 
@@ -65,6 +65,7 @@ jobs:
       pip install git+https://github.com/microsoft/onnxconverter-common
       pip install -r requirements.txt
       pip install -r requirements-dev.txt
+      pip install flatbuffers
       $(INSTALL_ORT)
     displayName: 'Install dependencies'
 

diff --git a/.azure-pipelines/win32-CI-keras-applications-nightly.yml b/.azure-pipelines/win32-CI-keras-applications-nightly.yml
@@ -43,7 +43,7 @@ jobs:
         INSTALL_KERAS: pip install keras==2.3.1
         UNINSTALL_KERAS:
         INSTALL_TENSORFLOW: pip install tensorflow==1.14.0
-        INSTALL_ORT: pip install -i https://test.pypi.org/simple/ ort-nightly
+        INSTALL_ORT: pip install -i https://test.pypi.org/simple/ ort-nightly==1.8.0.dev202106023
         INSTALL_KERAS_RESNET: pip install keras-resnet
         INSTALL_TRANSFORMERS:
         NIGHTLY_BUILD_TEST: python run_all.py --exclude "test_keras_applications_v2.py test_mask_rcnn.py"
@@ -65,7 +65,7 @@ jobs:
         INSTALL_KERAS:
         UNINSTALL_KERAS: pip uninstall keras -y
         INSTALL_TENSORFLOW: pip install tensorflow==2.2.0
-        INSTALL_ORT: pip install onnxruntime==1.7.0
+        INSTALL_ORT: pip install -i https://test.pypi.org/simple/ ort-nightly==1.8.0.dev202106023
         INSTALL_KERAS_RESNET: pip install keras-resnet
         INSTALL_TRANSFORMERS: pip install transformers==3.4.0
         NIGHTLY_BUILD_TEST: python run_all_v2.py
@@ -95,6 +95,7 @@ jobs:
       pip install git+https://github.com/microsoft/onnxconverter-common
       pip install -r requirements.txt
       pip install -r requirements-dev.txt
+      pip install flatbuffers
       %INSTALL_ORT%
       pip install opencv-python
       pip install tqdm

diff --git a/.azure-pipelines/win32-conda-CI-tf-keras.yml b/.azure-pipelines/win32-conda-CI-tf-keras.yml
@@ -42,7 +42,7 @@ jobs:
         python.version: '3.8'
         ONNX_PATH: onnx==1.8.0
         TENSORFLOW_PATH: tensorflow-cpu==2.3.0
-        INSTALL_ORT: pip install onnxruntime==1.7.0
+        INSTALL_ORT: pip install -i https://test.pypi.org/simple/ ort-nightly==1.8.0.dev202106023
 
     maxParallel: 3
 
@@ -68,6 +68,7 @@ jobs:
       pip install git+https://github.com/microsoft/onnxconverter-common
       pip install -r requirements.txt
       pip install -r requirements-dev.txt
+      pip install flatbuffers
       %INSTALL_ORT%
     displayName: 'Install dependencies'