diff --git a/requirement.txt b/requirement.txt
index bf7425f..af3cbdd 100644
--- a/requirement.txt
+++ b/requirement.txt
@@ -1,7 +1,7 @@
 absl-py==0.7.1
 astor==0.7.1
 atari-py==0.1.7
-tensorflow-gpu==1.12.0
+tensorflow-gpu==2.6.4
 certifi==2019.3.9
 cffi==1.12.3
 chardet==3.0.4