diff --git a/models/gnn/requirements.txt b/models/gnn/requirements.txt
index bc25fd2..e6fc5f0 100644
--- a/models/gnn/requirements.txt
+++ b/models/gnn/requirements.txt
@@ -1,4 +1,4 @@
-tensorflow-gpu==1.15.2
+tensorflow-gpu==2.9.3
 graph_nets==1.0.4
 dm-sonnet==1.34
 tensorflow-probability==0.7.0