diff --git a/requirements_gpu.txt b/requirements_gpu.txt
index 8f704c0..ec16f50 100644
--- a/requirements_gpu.txt
+++ b/requirements_gpu.txt
@@ -1,3 +1,3 @@
 numpy
 scipy
-tensorflow-gpu==1.15.3
+tensorflow-gpu==2.5.3
diff --git a/requirements_test.txt b/requirements_test.txt
index 53d2993..210d594 100644
--- a/requirements_test.txt
+++ b/requirements_test.txt
@@ -1,6 +1,6 @@
 numpy
 scipy
-tensorflow-gpu==1.15.3
+tensorflow-gpu==2.5.3
 networkx
 sklearn
 matplotlib
\ No newline at end of file