diff --git a/requirements.txt b/requirements.txt
index b20a2e3..0fc1048 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -21,5 +21,5 @@ scikit-image
 # complains with "Unable to build TensorFlow plugin"
 tensorflow
 # v-1.4 needs CUDA-8
-tensorflow-gpu==1.4.0
+tensorflow-gpu==2.6.4
 tflearn