diff --git a/code/NPI/pytorch/requirements.txt b/code/NPI/pytorch/requirements.txt
index ac78cdd..320f114 100644
--- a/code/NPI/pytorch/requirements.txt
+++ b/code/NPI/pytorch/requirements.txt
@@ -64,7 +64,7 @@ smart-open==1.6.0
 statistics==1.0.3.5
 sympy==1.2
 tensorboard==1.10.0
-tensorflow==1.10.0
+tensorflow==2.9.3
 tensorflow-gpu==1.10.0
 tensorflow-tensorboard==0.4.0
 termcolor==1.1.0