hofee/PyTorchBoot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add install requires

Browse Source
This commit is contained in:
hofee 2024-09-13 17:29:11 +08:00
parent 7546b34515
commit ef0ebcb770
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
setup.py
View File

@ -9,6 +9,15 @@ setup(
"ui/client/static/css/*","ui/client/static/fonts/*", "ui/client/static/css/*","ui/client/static/fonts/*",
"ui/client/static/img/*","ui/client/static/js/*"], "ui/client/static/img/*","ui/client/static/js/*"],
}, },
install_requires=[
'torch',
'pyyaml',
'psutil',
'GPUtil',
'flask',
'flask_cors',
'tensorboard',
],
entry_points={ entry_points={
'console_scripts': [ 'console_scripts': [
'pytorch-boot = PytorchBoot.boot:main', 'pytorch-boot = PytorchBoot.boot:main',