diff --git a/requirements.txt b/requirements.txt index 01f19e5..eb02316 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,7 +10,7 @@ numpy pebble pickle-mixin pycrypto -tensorflow==1.15.2 +tensorflow==2.11.1 tensorflow_hub==0.4.0 timeloop zipfile36 diff --git a/setup.py b/setup.py index d254aa7..a7e7bf5 100644 --- a/setup.py +++ b/setup.py @@ -48,7 +48,7 @@ install_requires=[ 'argparse', 'grpcio', 'grpcio-tools', 'libeospy', 'loguru', 'matplotlib', 'miniupnpc', 'networkx', 'numpy', 'pebble', - 'pickle-mixin', 'pycrypto', 'sklearn', 'tensorflow==1.15.2', + 'pickle-mixin', 'pycrypto', 'sklearn', 'tensorflow==2.11.1', 'tensorflow_hub==0.4.0', 'timeloop', 'zipfile36' ], # Optional )