diff --git a/setup.py b/setup.py
index b45596d..637dd69 100644
--- a/setup.py
+++ b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import find_packages
 from setuptools import setup
 
-REQUIRED_PACKAGES = ['Keras==2.2.4', 'tensorflow==1.14', 'numpy==1.16.5']
+REQUIRED_PACKAGES = ['Keras==2.2.4', 'tensorflow==2.5.3', 'numpy==1.16.5']
 
 setup(
     name='trainer',
diff --git a/trainer/requirements.txt b/trainer/requirements.txt
index c1a4ba1..84db8a5 100644
--- a/trainer/requirements.txt
+++ b/trainer/requirements.txt
@@ -1,4 +1,4 @@
 numpy>=1.14
 pandas>=0.22
 six>=1.11
-tensorflow>=1.13,<2
+tensorflow>=1.13,<3