Added missing dependencies and switched to GPU dependency

Signed-off-by: Jim Martens <github@2martens.de>
2019-02-07 16:58:32 +01:00 · 2019-02-07 16:58:32 +01:00 · bfc888389f
parent ecbfbf2a2a
commit bfc888389f
1 changed files with 1 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@ -35,7 +35,7 @@ setup(
    package_data={},
    entry_points={},
    python_requires="~=3.6",
-    install_requires=["tensorflow"],
+    install_requires=["tensorflow-gpu", "Pillow", "h5py", "numpy"],
    license="Apache License 2.0",
    classifiers=[
        "Operating System :: OS Independent",