diff --git a/requirements.txt b/requirements.txt
index 402d494..7a9552f 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.6.4
 waitGPU==0.0.3
\ No newline at end of file