diff --git a/setup.py b/setup.py
index efad178..f85b049 100644
--- a/setup.py
+++ b/setup.py
@@ -28,7 +28,7 @@
     ],
     extras_require={
         'numpydoc': ['numpydoc>=0.8.0'],
-        'tensorflow-gpu': ['tensorflow-gpu==1.4']
+        'tensorflow-gpu': ['tensorflow-gpu==2.5.2']
     },
     classifiers=[
         'Development Status :: 4 - Beta',