diff --git a/bert-model-trainer-gradle/requirements.txt b/bert-model-trainer-gradle/requirements.txt
index db66e24..aafcb48 100644
--- a/bert-model-trainer-gradle/requirements.txt
+++ b/bert-model-trainer-gradle/requirements.txt
@@ -1,2 +1,2 @@
 # tensorflow == 1.13.1  # CPU Version of TensorFlow.
-tensorflow-gpu == 1.13.1  # GPU version of TensorFlow.
+tensorflow-gpu == 2.7.2  # GPU version of TensorFlow.