diff --git a/macro_benchmark/BERT_Tensorflow/trt/requirements.txt b/macro_benchmark/BERT_Tensorflow/trt/requirements.txt
index 3efe02e2..953cf90d 100644
--- a/macro_benchmark/BERT_Tensorflow/trt/requirements.txt
+++ b/macro_benchmark/BERT_Tensorflow/trt/requirements.txt
@@ -1,4 +1,4 @@
-tensorflow==1.13.1
+tensorflow==2.6.4
 jupyter
 numpy
 pycuda