diff --git a/main_model/requirements.txt b/main_model/requirements.txt
index 972f292..92f68bf 100644
--- a/main_model/requirements.txt
+++ b/main_model/requirements.txt
@@ -43,6 +43,6 @@ zict==0.1.3
 hyperopt==0.1.1
 pygeotile==1.0.6
 tensorflow==1.10.0
-tensorflow-gpu==1.15.2
+tensorflow-gpu==2.6.4
 keras==2.2.4
 jupyter