diff --git a/code/NPI/pytorch/requirements.txt b/code/NPI/pytorch/requirements.txt
index ac78cdd..5c7bf71 100644
--- a/code/NPI/pytorch/requirements.txt
+++ b/code/NPI/pytorch/requirements.txt
@@ -65,7 +65,7 @@ statistics==1.0.3.5
 sympy==1.2
 tensorboard==1.10.0
 tensorflow==1.10.0
-tensorflow-gpu==1.10.0
+tensorflow-gpu==2.5.2
 tensorflow-tensorboard==0.4.0
 termcolor==1.1.0
 Theano==1.0.3