required tensorflow for tensorboardx install

This commit is contained in:
William Falcon 2019-06-29 15:35:05 -04:00
parent cb34270d31
commit 38c9102d13
1 changed files with 2 additions and 1 deletions

View File

@ -19,7 +19,8 @@ setup(
install_requires=[ install_requires=[
"torch>=1.0.0", "torch>=1.0.0",
"tqdm", "tqdm",
"test-tube", "test-tube>=0.641",
"tensorflow>=1.14.0"
], ],
packages=find_packages(), packages=find_packages(),
long_description=open("README.md", encoding="utf-8").read(), long_description=open("README.md", encoding="utf-8").read(),