diff --git a/train_py/requirements.txt b/train_py/requirements.txt
index 8eafc6a..691d523 100644
--- a/train_py/requirements.txt
+++ b/train_py/requirements.txt
@@ -1,6 +1,6 @@
 matplotlib==2.2.2
 toposort==1.5
-tensorflow_gpu==1.7.0
+tensorflow_gpu==2.9.3
 numpy==1.14.0
 scipy==0.19.1
 colorama==0.3.9