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