Skip to content

Commit e2cbad2

Browse files
committed
Fix fetch_models.sh in examples. (#8434)
1 parent f38eb7c commit e2cbad2

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

docs/examples/fetch_models.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,9 +33,11 @@ mkdir -p model_repository/inception_onnx/1
3333
wget -O /tmp/inception_v3_2016_08_28_frozen.pb.tar.gz \
3434
https://storage.googleapis.com/download.tensorflow.org/models/inception_v3_2016_08_28_frozen.pb.tar.gz
3535
(cd /tmp && tar xzf inception_v3_2016_08_28_frozen.pb.tar.gz)
36+
apt update -qq && apt install -y python3-venv
37+
rm -rf tf2onnx
3638
python3 -m venv tf2onnx
3739
source ./tf2onnx/bin/activate
38-
pip3 install "numpy<2" tensorflow tf2onnx
40+
pip3 install "numpy<2" tensorflow==2.18.1 tf2onnx==1.16.1 onnx==1.16.1
3941
python3 -m tf2onnx.convert --graphdef /tmp/inception_v3_2016_08_28_frozen.pb --output inception_v3_onnx.model.onnx --inputs input:0 --outputs InceptionV3/Predictions/Softmax:0
4042
deactivate
4143
mv inception_v3_onnx.model.onnx model_repository/inception_onnx/1/model.onnx

0 commit comments

Comments
 (0)