diff --git a/bert-model-trainer-gradle/requirements.txt b/bert-model-trainer-gradle/requirements.txt
index db66e24..552d710 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.12.0  # GPU version of TensorFlow.