Added missing dependencies and switched to GPU dependency

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-02-07 16:58:32 +01:00
parent ecbfbf2a2a
commit bfc888389f

View File

@ -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",