diff --git a/setup.py b/setup.py
index c34cec9d..4ee11496 100644
--- a/setup.py
+++ b/setup.py
@@ -70,7 +70,7 @@ def find_version():
         'interactive': ['matplotlib==3.0.3'],
         'none': [],
         'cpu': ['tensorflow==1.13.2'],
-        'gpu': ['tensorflow-gpu==1.13.2'],
+        'gpu': ['tensorflow-gpu==2.5.0'],
         'dev': [
             'autopep8>=1.5',
             'flake8>=3.7.9',