general/occlum
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix tensorflw serving error

Browse Source
This commit is contained in:
Hui, Chunyang 2023-09-18 09:41:39 +00:00 committed by volcano
parent 75dea2d43c
commit 775350bd4e
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
demos/tensorflow/tensorflow_serving/client/requirements.txt
View File

@ -2,5 +2,7 @@ argparse
aiohttp>=3.7.0 aiohttp>=3.7.0
grpcio>=1.34.0 grpcio>=1.34.0
opencv-python>=4.4.0 opencv-python>=4.4.0
tensorflow>=2.3.0 tensorflow==2.13
tensorflow-serving-api>=2.3.0 tensorflow-serving-api==2.13
protobuf==3.20.3
numpy==1.22.1

2
demos/tensorflow/tensorflow_serving/prepare_model_and_env.sh
View File

@ -9,6 +9,6 @@ else
fi fi
pip3 install --upgrade pip 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 ./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 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