diff --git a/imperative/python/megengine/__init__.py b/imperative/python/megengine/__init__.py index f71c96b96a5815d9047157238ee272891470f084..8fd9e3bd13e885fd88e2ca759121e88abe09ca5c 100644 --- a/imperative/python/megengine/__init__.py +++ b/imperative/python/megengine/__init__.py @@ -1,4 +1,36 @@ # -*- coding: utf-8 -*- +filter_package_name = 'megbrain' +need_package_name = 'megengine' + + +class MegEngineBorkenEnv(Exception): + pass + + +import pkg_resources + +installed_packages = pkg_resources.working_set +installed_packages_list = sorted(["%s" % i.key.lower() for i in installed_packages]) +is_broken_env = False +if filter_package_name in installed_packages_list: + is_broken_env = True + print( + "broken env, please do uninstall {} and {}, then install {} again".format( + filter_package_name, need_package_name, need_package_name + ) + ) + +if "megenginelite" in installed_packages_list: + is_broken_env = True + print( + "broken env, please do uninstall megenginelite and {}, then install {} again".format( + need_package_name, need_package_name + ) + ) + +if is_broken_env: + raise MegEngineBorkenEnv + import atexit import ctypes import re diff --git a/imperative/python/requires.txt b/imperative/python/requires.txt index d08566d3f7ef84629ef3cb5e80a5a75ab4e2ba17..e9e457d7d1672de4dbfaccafed6516437d3b731d 100644 --- a/imperative/python/requires.txt +++ b/imperative/python/requires.txt @@ -9,4 +9,5 @@ mprop wheel megfile>=0.0.10 filelock -psutil \ No newline at end of file +psutil +setuptools