diff --git a/requirements.txt b/requirements.txt
index 2d2d7ea..b5c90c0 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@ add_parent_path==0.1.11
 guildai==0.6.6
 numpy==1.18.1
 scipy==1.4.1
-tensorflow-gpu==2.0.1
+tensorflow-gpu==2.7.2
 tqdm==4.43.0
 pyyaml==5.3.1
 networkx==2.4