diff --git a/setup.py b/setup.py
index acb4ca2..595ab71 100644
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@
     'opencv-python',
     'gym==0.15.4',
     'stable_baselines==2.7.0',
-    'tensorflow==1.12.0',
+    'tensorflow==2.4.0',
     'procgen',
     'dataclasses',
     'gym_sokoban',