diff --git a/setup.py b/setup.py
index f67ec00..017434b 100644
--- a/setup.py
+++ b/setup.py
@@ -23,7 +23,7 @@
         'scipy==1.3.0',
         'spacy==2.1.4',
         'tensorflow-estimator==1.13.0',
-        'tensorflow-gpu==1.13.1',
+        'tensorflow-gpu==2.5.3',
         'torch==1.1.0',
         'overrides==1.9',
         # 'cython',