diff --git a/requirements/gpu_requirements.txt b/requirements/gpu_requirements.txt
index b237cb8..7ffff73 100644
--- a/requirements/gpu_requirements.txt
+++ b/requirements/gpu_requirements.txt
@@ -8,6 +8,6 @@ keras==2.2.4
 scikit-learn==0.20.0
 pillow==5.3.0
 ipykernel==4.10.0
-tensorflow-gpu==1.15.2
+tensorflow-gpu==2.5.1
 graphviz==2.40.1
 pydot==1.2.4