From 6cf209ed68c051938bf59cfa3b0722cf09bda289 Mon Sep 17 00:00:00 2001 From: krishung5 Date: Tue, 5 Sep 2023 16:47:53 -0700 Subject: [PATCH] Upgrade protobuf version --- qa/common/gen_qa_model_repository | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/qa/common/gen_qa_model_repository b/qa/common/gen_qa_model_repository index 246749c093..85db0730b2 100755 --- a/qa/common/gen_qa_model_repository +++ b/qa/common/gen_qa_model_repository @@ -386,6 +386,10 @@ nvidia-smi -L || true nvidia-smi || true set -e set -x +# Segmentation fault with protobuf 4.24.0 (https://github.com/tensorflow/tensorflow/issues/61551) +# Upgrade protobuf version to fix the issue. +pip3 install "protobuf>4.24.0" + python3 $SRCDIR/gen_qa_models.py --graphdef --savedmodel --models_dir=$DESTDIR chmod -R 777 $DESTDIR python3 $SRCDIR/gen_qa_models.py --graphdef --savedmodel --variable --models_dir=$VARDESTDIR