Open burningion opened 5 months ago
Following up here, I tried using the container from nanoowl, and although trtexec
wasn't on the PATH, it was in /usr/src/tensorrt/bin/trtexec
. Using that I was able to successfully run optimization.
Commands became:
$ /usr/src/tensorrt/bin/trtexec --onnx=data/mobile_sam_mask_decoder.onnx --saveEngine=data/mobile_sam_mask_decoder.engine --minShapes=point_coords:1x1x2,point_labels:1x1 --optShapes=point_coords:1x1x2,point_labels:1x1 --maxShapes=point_coords:1x10x2,point_labels:1x10
...
$ /usr/src/tensorrt/bin/trtexec --onnx=data/resnet18_image_encoder.onnx --saveEngine=data/resnet18_image_encoder.engine --fp16
Hey there,
Thanks for the work on nanosam! I'm working on a project, and trying to run nanosam from the included Docker image.
I've added the line:
To the
docker/23-01/run.sh
so it'll run on Jetpack 6 Developer Preview.When I add the
resnet18_image_encoder.onnx
and themobile_sam_mask_decoder.onnx
for training, I get the same error:It appears both of these have the same CUDA driver version is insufficient.
When I try to do a:
In the container, it just fails:
Any tips for further debugging? Should we rebuild containers for Jetpack 6?