Get tvmc version from tvm (apache#7478)

Change-Id: I6a6e78080f36e4e3e1689e03ea48e759fcd8e466
neo-ai · Mar 2, 2021 · 6c1434f · 6c1434f
1 parent 2196b41
commit 6c1434f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/python/tvm/driver/tvmc/main.py b/python/tvm/driver/tvmc/main.py
@@ -23,7 +23,7 @@
 import logging
 import sys
 
-import pkg_resources
+import tvm
 
 from tvm.driver.tvmc.common import TVMCException
 
@@ -75,8 +75,7 @@ def _main(argv):
     logging.getLogger("TVMC").setLevel(40 - args.verbose * 10)
 
     if args.version:
-        version = pkg_resources.get_distribution("tvm").version
-        sys.stdout.write("%s\n" % version)
+        sys.stdout.write("%s\n" % tvm.__version__)
         return 0
 
     if not hasattr(args, "func"):