diff --git a/requirements.txt b/requirements.txt
index efb6fe0..c4a3f02 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
 numpy==1.17.2
-tensorflow==2.0.0
+tensorflow==2.6.4
 gym
 falcon
diff --git a/setup.py b/setup.py
index cc69914..0db4766 100644
--- a/setup.py
+++ b/setup.py
@@ -4,6 +4,6 @@
     name = 'policygradient',
     version = 0.1,
     packages = find_packages(),
-    install_requires = ['pip >= 19.3.1', 'numpy', 'tensorflow==2.0.0', 'gym', 'falcon'],
+    install_requires = ['pip >= 19.3.1', 'numpy', 'tensorflow==2.6.4', 'gym', 'falcon'],
     include_package_data = True
 )