diff --git a/replab_grasping/training/requirements.txt b/replab_grasping/training/requirements.txt
index 2daa956..06c9d26 100644
--- a/replab_grasping/training/requirements.txt
+++ b/replab_grasping/training/requirements.txt
@@ -5,7 +5,7 @@ scikit-learn==0.20.3
 scipy==1.2.1
 tensorboard==1.13.1
 tensorboardX==1.6
-tensorflow==1.13.1
+tensorflow==2.3.1
 tensorflow-estimator==1.13.0
 torch==1.0.1.post2
 torchvision==0.2.2.post3