Browse Source

feat(whl): check broken env before import

GitOrigin-RevId: 31ddb2a8c2
master
Megvii Engine Team 2 years ago
parent
commit
d733f429d9
2 changed files with 34 additions and 1 deletions
  1. +32
    -0
      imperative/python/megengine/__init__.py
  2. +2
    -1
      imperative/python/requires.txt

+ 32
- 0
imperative/python/megengine/__init__.py View File

@@ -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


+ 2
- 1
imperative/python/requires.txt View File

@@ -9,4 +9,5 @@ mprop
wheel
megfile>=0.0.10
filelock
psutil
psutil
setuptools

Loading…
Cancel
Save