diff --git a/pernaf2/pernaf.egg-info/requires.txt b/pernaf2/pernaf.egg-info/requires.txt
index 4b673af..0dcc07e 100644
--- a/pernaf2/pernaf.egg-info/requires.txt
+++ b/pernaf2/pernaf.egg-info/requires.txt
@@ -1,3 +1,3 @@
-tensorflow==1.14
+tensorflow==2.5.3
 numpy
 gym
diff --git a/pernaf2/setup.py b/pernaf2/setup.py
index 0492acb..2c3ca27 100644
--- a/pernaf2/setup.py
+++ b/pernaf2/setup.py
@@ -47,7 +47,7 @@ def reqs(*f):
     url = 'https://github.com/MathPhysSim/PER-NAF',
     long_description=long_description,
     long_description_content_type='text/markdown',
-    install_requires=['tensorflow==1.14', 'numpy', 'gym'],
+    install_requires=['tensorflow==2.5.3', 'numpy', 'gym'],
     packages=find_packages(),
     include_package_data=True,
     classifiers=[