diff --git a/requirements.txt b/requirements.txt index 673406b..7ae1e03 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ pillow==6.2.1 # 7.0.0 breaks torchvision numpy==1.16.3 torch==1.0.1 torchvision==0.2.2.post3 -tensorflow==1.8.0 # works on dgx not on Mac... +tensorflow==2.9.3 # works on dgx not on Mac... # tensorflow==1.14.0 # works on Mac not on dgx... tensorboardX==1.6 neo4jrestclient==2.1.1 diff --git a/setup.py b/setup.py index e05a51e..33b9409 100644 --- a/setup.py +++ b/setup.py @@ -19,7 +19,7 @@ py_modules=['smtag.command_line'], install_requires=[ #'tensorflow==1.8.0', # dgx - 'tensorflow==1.14.0', # mac needed for tensorboardX visualization + 'tensorflow==2.9.3', # mac needed for tensorboardX visualization 'tensorboardX', 'docopt', 'torch==1.0.1', diff --git a/smtag/requirements.txt b/smtag/requirements.txt index c35207d..bdc2d32 100644 --- a/smtag/requirements.txt +++ b/smtag/requirements.txt @@ -1,5 +1,5 @@ # tensorflow==1.8.0 # works on dgx not on Mac... -tensorflow==1.14.0 # works on Mac not on dgx... +tensorflow==2.9.3 # works on Mac not on dgx... docopt==0.6.2 numpy==1.16.3 torch==1.0.1