diff --git a/x2paddle/convert.py b/x2paddle/convert.py index 5ef50657e214fbbd17d3429a2a3e91bac4a8b696..1dc83e55af5cc8105520baf2123499370e11ae1f 100644 --- a/x2paddle/convert.py +++ b/x2paddle/convert.py @@ -142,12 +142,12 @@ def onnx2paddle(model_path, save_dir): try: import torch version = torch.__version__ - if '1.1.0' not in version: - print("torch==1.1.0 is required") + if '1.2.0' not in version: + print("torch==1.2.0 is required") return except: print( - "we use caffe2 to inference graph, please use \"pip install torch==1.1.0\"." + "we use caffe2 to inference graph, please use \"pip install torch==1.2.0\"." ) return try: @@ -190,14 +190,6 @@ def main(): x2paddle.__version__)) return - try: - import paddle - v0, v1, v2 = paddle.__version__.split('.') - if int(v0) != 1 or int(v1) < 5: - print("paddlepaddle>=1.5.0 is required") - return - except: - print("paddlepaddle not installed, use \"pip install paddlepaddle\"") assert args.framework is not None, "--framework is not defined(support tensorflow/caffe/onnx)" assert args.save_dir is not None, "--save_dir is not defined"