diff --git a/macro_benchmark/BERT_Tensorflow/trt/requirements.txt b/macro_benchmark/BERT_Tensorflow/trt/requirements.txt index 3efe02e2..e110ecbc 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.5.3 jupyter numpy pycuda