File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -430,7 +430,7 @@ python3 $VOLUME_SRCDIR/gen_qa_ragged_models.py --tensorrt --models_dir=$VOLUME_R
430
430
chmod -R 777 $VOLUME_RAGGEDDESTDIR
431
431
python3 $VOLUME_SRCDIR /gen_qa_trt_format_models.py --models_dir=$VOLUME_FORMATDESTDIR
432
432
chmod -R 777 $VOLUME_FORMATDESTDIR
433
- python3 $VOLUME_SRCDIR /gen_qa_trt_data_dependent_shape.py --models_dir=$VOLUME_DATADEPENDENTDIR
433
+ nvidia-smi --query-gpu=compute_cap | grep -qz 11.0 && echo -e '\033[33m[WARNING]\033[0m Skipping model generation for data dependent shape' || python3 $VOLUME_SRCDIR /gen_qa_trt_data_dependent_shape.py --models_dir=$VOLUME_DATADEPENDENTDIR
434
434
chmod -R 777 $VOLUME_DATADEPENDENTDIR
435
435
# Make shared library for custom Hardmax plugin.
436
436
if [ -d "/usr/src/tensorrt" ]; then
You can’t perform that action at this time.
0 commit comments