Fix version check bug (#6784)

* Fix version check bug * Update pytorch_utils.py * Update pytorch_utils.py * Update pytorch_utils.py * Update pytorch_utils.py
apache · Oct 28, 2020 · 99027fc · 99027fc
1 parent b4858d4
commit 99027fc
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]
+    )