ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 00:56:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add package install options, pip install trains[gpu,s3,azure,gs]

Browse Source
This commit is contained in:
allegroai 2019-09-03 12:59:32 +03:00
parent 64ba30df13
commit 4d8955dfd3
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
setup.py
View File

@ -62,6 +62,20 @@ setup(
'deeplearning deep-learning experiment-manager experimentmanager',
packages=find_packages(exclude=['contrib', 'docs', 'data', 'examples', 'tests']),
install_requires=requirements,
extras_require={
's3': [
'boto3>=1.9',
],
'azure': [
'azure.storage.blob>=2.0.1',
],
'gs': [
'google-cloud-storage>=1.13.2',
],
'gpu': [
'gpustat',
],
},
package_data={
'trains': ['config/default/*.conf', 'backend_api/config/default/*.conf']
},