提交 41a6297d 编写于 作者: C channingss

add code for check onnxruntime version

上级 b0a7761a
import onnxruntime as rt
import os
import sys
import numpy as np
......@@ -19,6 +18,17 @@ def arg_parser():
def main():
try:
import onnxruntime as rt
version = rt.__version__
if version != '0.4.0':
print("onnxruntime==0.4.0 is required")
return
except:
print(
"onnxruntime is not installed, use \"pip install onnxruntime==0.4.0\"."
)
return
parser = arg_parser()
args = parser.parse_args()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册