diff --git a/requirements.txt b/requirements.txt
index ae3efb2..be97b8e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 torch==1.0
 torchvision
-tensorflow==1.15.2
+tensorflow==2.6.4
 tensorboardX
 gym
 gym[atari]