diff --git a/setup.py b/setup.py
index d3544f5..1ce0151 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 from setuptools import find_packages
 from setuptools import setup
 
-REQUIRED_PACKAGES = ['tensorflow-gpu==2.0.0',
+REQUIRED_PACKAGES = ['tensorflow-gpu==2.7.2',
                      'fastprogress==0.1.21',
                      'pandas==0.25.0',
                      'transformers==2.1.1']