required tensorflow for tensorboardx install

2019-06-29 15:35:05 -04:00 · 2019-06-29 15:35:05 -04:00 · 38c9102d13
parent cb34270d31
commit 38c9102d13
1 changed files with 2 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@ -19,7 +19,8 @@ setup(
    install_requires=[
        "torch>=1.0.0",
        "tqdm",
-        "test-tube",
+        "test-tube>=0.641",
+        "tensorflow>=1.14.0"
    ],
    packages=find_packages(),
    long_description=open("README.md", encoding="utf-8").read(),