diff --git a/requirements.txt b/requirements.txt
index 7684d53..cb83d84 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 imageio>=2.5.0
 numpy
-tensorflow_gpu==1.15.2
+tensorflow_gpu==2.6.4
 Pillow>=6.1.0