diff --git a/training/requirements.txt b/training/requirements.txt
index 37ebe05..044ac2e 100644
--- a/training/requirements.txt
+++ b/training/requirements.txt
@@ -1,4 +1,4 @@
-tensorflow-gpu==1.4.0
+tensorflow-gpu==2.3.1
 matplotlib==2.1.0
 numpy==1.13.3
 scipy==1.0.0