Update setup.py

This commit is contained in:
“阅谦” 2019-07-21 13:11:24 +08:00
parent 1be09be85b
commit 1d3728e1f7

View File

@ -17,7 +17,6 @@ REQUIRED_PACKAGES = [
'scikit-learn',
'pandas',
'matplotlib',
'tensorflow'
]
@ -41,9 +40,9 @@ setuptools.setup(
extras_require={
"tf": ['tensorflow>=1.4.0,!=1.7.*,!=1.8.*'],
"cpu": ['tensorflow>=1.4.0,!=1.7.*,!=1.8.*'],
"tf_gpu": ['tensorflow-gpu>=1.4.0,!=1.7.*,!=1.8.*'],
"gpu": ['tensorflow-gpu>=1.4.0,!=1.7.*,!=1.8.*'],
},