diff --git a/requirements.txt b/requirements.txt
index 8be6de0..a400f83 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@ absl-py==0.7.0
 numpy==1.16.4
 scipy==1.2.0
 tensorflow==1.15.0
-tensorflow-gpu==1.15.0
+tensorflow-gpu==2.7.2
 tensorflow_probability==0.7.0
 matplotlib
 scikit-learn