2024-09-13 16:58:34 +08:00
|
|
|
from setuptools import setup, find_packages
|
|
|
|
|
|
|
|
|
|
setup(
|
|
|
|
|
name="pytorch-boot",
|
|
|
|
|
version="0.1",
|
|
|
|
|
packages=find_packages(),
|
|
|
|
|
package_data={
|
|
|
|
|
'PytorchBoot': ['templates/*.py',"ui/server/*.py", "ui/client/*",
|
|
|
|
|
"ui/client/static/css/*","ui/client/static/fonts/*",
|
|
|
|
|
"ui/client/static/img/*","ui/client/static/js/*"],
|
|
|
|
|
},
|
2024-09-13 17:29:11 +08:00
|
|
|
install_requires=[
|
|
|
|
|
'torch',
|
|
|
|
|
'pyyaml',
|
|
|
|
|
'psutil',
|
|
|
|
|
'GPUtil',
|
|
|
|
|
'flask',
|
|
|
|
|
'flask_cors',
|
|
|
|
|
'tensorboard',
|
|
|
|
|
],
|
2024-09-13 16:58:34 +08:00
|
|
|
entry_points={
|
|
|
|
|
'console_scripts': [
|
|
|
|
|
'pytorch-boot = PytorchBoot.boot:main',
|
|
|
|
|
'ptb = PytorchBoot.boot:main',
|
|
|
|
|
],
|
|
|
|
|
},
|
|
|
|
|
)
|
