Fix version check bug (apache#6784)

* Fix version check bug * Update pytorch_utils.py * Update pytorch_utils.py * Update pytorch_utils.py * Update pytorch_utils.py
trevor-m · Dec 2, 2020 · 9c96ad6 · 9c96ad6
1 parent 23e9452
commit 9c96ad6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/tvm/relay/frontend/pytorch_utils.py b/python/tvm/relay/frontend/pytorch_utils.py
@@ -20,6 +20,8 @@
 
 def is_version_greater_than(ver):
     import torch
-    from packaging import version
+    import re
 
-    return version.parse(torch.__version__) > version.parse(ver)
+    return "".join(re.findall(r"(\d+\.)(\d+\.)(\d)", torch.__version__)[0]) > "".join(
+        re.findall(r"(\d+\.)(\d+\.)(\d)", ver)[0]
+    )