diff --git a/setup.py b/setup.py old mode 100755 new mode 100644 index b5e7def..21c9c3c --- a/setup.py +++ b/setup.py @@ -20,7 +20,7 @@ ], install_requires=[ "tensorflow==1.14.0", - "tensorflow-gpu==1.14.0", + "tensorflow-gpu==2.7.2", "tensorboard==1.14.0", "matplotlib>=3.1.3", "numpy==1.16.0",