diff --git a/requirements.txt b/requirements.txt
index 402d494..38b6ae3 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -20,5 +20,5 @@ gurobipy==5.0.2
 Pillow==7.1.2
 posix_ipc==1.0.4
 setGPU==0.0.7
-tensorflow-gpu==1.15.4
+tensorflow-gpu==2.5.3
 waitGPU==0.0.3
\ No newline at end of file