diff --git a/docker/requirements.txt b/docker/requirements.txt
index a63d65c..cfe55ed 100644
--- a/docker/requirements.txt
+++ b/docker/requirements.txt
@@ -1,4 +1,4 @@
-tensorflow-gpu==1.4.0
+tensorflow-gpu==2.9.3
 keras==2.1.3
 keras_applications==1.0.7
 opencv-python==3.4.0.14
diff --git a/requirements.txt b/requirements.txt
index 29e7296..553b0e8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 opencv-python==3.4.0.14
 opencv-contrib-python==3.4.2.17
-tensorflow_gpu==1.4.0
+tensorflow_gpu==2.9.3
 matplotlib
 scikit-learn
 torch==0.4.1