diff --git a/requirements.txt b/requirements.txt
index 48bd83d..4abdc22 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 opencv-python>=4.1
 tensorpack==0.9.8
-tensorflow-gpu==1.15.2
+tensorflow-gpu==2.6.4
 python-prctl>=1.7
 tqdm>=4.36.1
 cython>=0.29.13