We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c40a3f5 commit c50245cCopy full SHA for c50245c
1 file changed
qa/common/gen_qa_model_repository
@@ -326,6 +326,7 @@ cat > $TORCHSCRIPT <<EOF
326
umask 0000
327
nvidia-smi --query-gpu=compute_cap,compute_mode,driver_version,name,index --format=csv || true
328
nvidia-smi || true
329
+pip3 install onnxscript
330
set -e
331
set -x
332
python3 $VOLUME_SRCDIR/gen_qa_models.py --libtorch --models_dir=$VOLUME_DESTDIR
0 commit comments