diff --git a/requirements.txt b/requirements.txt
index 6f84faa..a872022 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,7 +6,7 @@ torch==0.4.1
 torchvision==0.2.0 
 opencv-python
 pyyaml 
-tensorflow-gpu==1.9.0 
+tensorflow-gpu==2.3.1 
 tensorboard==1.9.0 
 scikit-image==0.14.1
 scikit-learn