diff --git a/demos/tensorflow/tensorflow_serving/client/requirements.txt b/demos/tensorflow/tensorflow_serving/client/requirements.txt index f422d19d..23b1ac7d 100644 --- a/demos/tensorflow/tensorflow_serving/client/requirements.txt +++ b/demos/tensorflow/tensorflow_serving/client/requirements.txt @@ -2,5 +2,7 @@ argparse aiohttp>=3.7.0 grpcio>=1.34.0 opencv-python>=4.4.0 -tensorflow>=2.3.0 -tensorflow-serving-api>=2.3.0 +tensorflow==2.13 +tensorflow-serving-api==2.13 +protobuf==3.20.3 +numpy==1.22.1 diff --git a/demos/tensorflow/tensorflow_serving/prepare_model_and_env.sh b/demos/tensorflow/tensorflow_serving/prepare_model_and_env.sh index f2a38dd6..fb077091 100755 --- a/demos/tensorflow/tensorflow_serving/prepare_model_and_env.sh +++ b/demos/tensorflow/tensorflow_serving/prepare_model_and_env.sh @@ -9,6 +9,6 @@ else fi pip3 install --upgrade pip -pip3 install --upgrade tensorflow==2.4 protobuf==3.19.2 +pip3 install --upgrade tensorflow==2.13 protobuf==3.20.3 numpy==1.22.1 ./download_model.sh python3 ./model_graph_to_saved_model.py --import_path ./models/resnet50-v15-fp32/resnet50-v15-fp32.pb --export_dir ./resnet50-v15-fp32 --model_version 1 --inputs input --outputs predict