diff --git a/setup.py b/setup.py index b6f78a7..56b0261 100644 --- a/setup.py +++ b/setup.py @@ -12,7 +12,7 @@ install_requires=[ "numpy == 1.16.5", "h5py == 2.9.0", - "tensorflow-gpu == 2.0.0", + "tensorflow-gpu == 2.9.3", "tqdm == 4.35.0", "scikit-learn == 0.21.3", "scipy == 1.3.1",