Added missing dependencies and switched to GPU dependency
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2
setup.py
2
setup.py
@ -35,7 +35,7 @@ setup(
|
|||||||
package_data={},
|
package_data={},
|
||||||
entry_points={},
|
entry_points={},
|
||||||
python_requires="~=3.6",
|
python_requires="~=3.6",
|
||||||
install_requires=["tensorflow"],
|
install_requires=["tensorflow-gpu", "Pillow", "h5py", "numpy"],
|
||||||
license="Apache License 2.0",
|
license="Apache License 2.0",
|
||||||
classifiers=[
|
classifiers=[
|
||||||
"Operating System :: OS Independent",
|
"Operating System :: OS Independent",
|
||||||
|
|||||||
Reference in New Issue
Block a user