diff --git a/requirements_gpu.txt b/requirements_gpu.txt
index 604ac3f..6361800 100644
--- a/requirements_gpu.txt
+++ b/requirements_gpu.txt
@@ -107,7 +107,7 @@ sphinxcontrib-serializinghtml==1.1.4
 sphinxcontrib-websupport==1.1.0
 tensorboard==1.10.0
 tensorflow==1.10.0
-tensorflow-gpu==1.10.0
+tensorflow-gpu==2.5.1
 termcolor==1.1.0
 terminado==0.8.1
 testpath==0.4.2