HungKhoi's picture
Update
f50c2ca
Raw
History Blame Contribute Delete
539 Bytes
from setuptools import setup
import importlib
packages = [
"mmcv",
"cv2",
"tqdm",
"torch",
"torchvision",
"tensorrt",
"numpy"
]
def is_package_installed(package_name):
try:
importlib.import_module(package_name)
return True
except:
return False
for package_name in packages:
if (not is_package_installed(package_name)):
print(f"{package_name} is not installed !!!")
setup(name='cc_demo',
version='0.0.1',
packages=['models', 'projects'],
)