diff --git a/requirements.txt b/requirements.txt
index a36a34d..435ffdb 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -17,7 +17,7 @@ progress
 h5py
 Cython==0.27
 pydensecrf==1.0rc3
-tensorflow-gpu==1.10.0
+tensorflow-gpu==2.6.4
 bokeh
 PrettyTable
 hyperopt==0.1.2