From be0f062ff3e24b08f9bb9b5e874dfbe0b9c3ec48 Mon Sep 17 00:00:00 2001
From: channingss <chen_linghchi@163.com>
Date: Thu, 8 Aug 2019 16:40:46 +0800
Subject: [PATCH] add check torch version

---
 x2paddle/decoder/onnx_decoder.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/x2paddle/decoder/onnx_decoder.py b/x2paddle/decoder/onnx_decoder.py
index 76167c1..bb271a9 100644
--- a/x2paddle/decoder/onnx_decoder.py
+++ b/x2paddle/decoder/onnx_decoder.py
@@ -461,6 +461,15 @@ class ONNXDecoder(object):
         """
         get dynamic shape from caffe2.backend
         """
+        try:
+            import torch
+            version = torch.__version__
+            if '1.1.0' not in version:
+                print("torch==1.1.0 is required")
+                return
+        except:
+            print("onnx is not installed, use \"pip install torch==1.1.0\".")
+            return
         from caffe2.python.onnx.backend import prepare
         shape = input_shapes[0]
         np_images = np.random.rand(shape[0], shape[1], shape[2],
-- 
GitLab