зеркало из https://github.com/microsoft/archai.git
64 строки
1.7 KiB
Python
64 строки
1.7 KiB
Python
# Copyright (c) Microsoft Corporation.
|
|
# Licensed under the MIT license.
|
|
|
|
import setuptools
|
|
|
|
with open("README.md", "r", encoding="utf_8") as f:
|
|
long_description = f.read()
|
|
|
|
install_requires = [
|
|
"coloredlogs>=15.0.1",
|
|
"datasets>=2.4.0",
|
|
"ftfy>=6.1.1",
|
|
"gorilla>=0.4.0",
|
|
"h5py>=3.7.0",
|
|
"hyperopt>=0.2.7",
|
|
"kaleido>=0.2.1",
|
|
"matplotlib>=3.5.3",
|
|
"nv-dllogger>=1.0.0",
|
|
"onnx==1.10.2",
|
|
"onnxruntime==1.10.0",
|
|
"overrides==3.1.0",
|
|
"plotly>=5.10.0",
|
|
"psutil>=5.9.1",
|
|
"pynvml>=11.4.1",
|
|
"pyunpack>=0.3",
|
|
"pyyaml==6.0",
|
|
"ray>=1.0.0",
|
|
"requests==2.25.1",
|
|
"runstats>=2.0.0",
|
|
"sacremoses>=0.0.53",
|
|
"scikit-learn>=1.0.2",
|
|
"seaborn>=0.11.2",
|
|
"send2trash>=1.8.0",
|
|
"statopt>=0.2",
|
|
"sympy>=1.10.1",
|
|
"tensorboard>=2.10.0",
|
|
"tensorwatch>=0.9.1",
|
|
"tokenizers>=0.12.1",
|
|
"tqdm>=4.64.0",
|
|
"transformers>=4.21.1",
|
|
]
|
|
|
|
setuptools.setup(
|
|
name="archai",
|
|
version="0.6.7",
|
|
description="Platform for Neural Architecture Search",
|
|
long_description=long_description,
|
|
long_description_content_type="text/markdown",
|
|
url="https://github.com/microsoft/archai",
|
|
license="MIT",
|
|
install_requires=install_requires,
|
|
packages=find_packages(),
|
|
include_package_data=True,
|
|
classifiers=[
|
|
"Intended Audience :: Science/Research",
|
|
"License :: OSI Approved :: MIT License",
|
|
"Development Status :: 5 - Production/Stable",
|
|
"Operating System :: OS Independent",
|
|
"Programming Language :: Python :: 3",
|
|
"Topic :: Software Development :: Libraries",
|
|
"Topic :: Software Development :: Libraries :: Python Modules",
|
|
],
|
|
)
|