浏览代码

add paddlepaddle version check

jiangjiajun 5 年之前
父节点
当前提交
63f88667f7
共有 1 个文件被更改,包括 5 次插入0 次删除
  1. 5 0
      paddlex/__init__.py

+ 5 - 0
paddlex/__init__.py

@@ -43,6 +43,11 @@ except:
         "[WARNING] pycocotools install: https://paddlex.readthedocs.io/zh_CN/develop/install.html#pycocotools"
     )
 
+import paddle
+if paddle.__version__ == '0.0.0':
+    print("[NOTICE]\tYou are using paddlepaddle with version= 0.0.0")
+elif paddle.__version__ < '1.8.4':
+    raise Exception("[ERROR] paddlepaddle-gpu or paddlepaddle >= 1.8.4 is required")
 import paddlehub as hub
 if hub.version.hub_version < '1.8.2':
     raise Exception("[ERROR] paddlehub >= 1.8.2 is required")