diff --git a/requirements.txt b/requirements.txt
index d5a7f79..0a291b3 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,5 +4,5 @@ open3d-python
 parse==1.12.0
 scikit-learn==0.20.3
 tensorboard==1.10
-tensorflow-gpu==1.10
+tensorflow-gpu==2.5.1
 tqdm==4.31.1