diff --git a/requirements.txt b/requirements.txt
index 9b59752..0fc278e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,5 +11,5 @@ pyyaml
 scikit-image
 scipy
 sklearn
-tensorflow-gpu==1.14
+tensorflow-gpu==2.6.4
 vtk