NanoOwl Generative AI example memory error AGX Orin

So I am trying to run the NanoOwl Generative AI example:

But I seem to be having an Out of memory error.
I am running it on the AGX Orin 32 GB.
When I run this example:

python3 owl_predict.py --prompt=“[an owl, a glove]” --threshold=0.1 --image_encoder_engine=…/data/owl_image_encoder_patch32.engine

I get this Memory error

torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 4973.43 GiB (GPU 0; 29.88 GiB total capacity; 647.78 MiB already allocated; 22.52 GiB free; 684.00 MiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation.

Hi @adventuredaisy,

Thanks for reaching out!

This seems like an unusually large amount of memory for the program to allocate.

Do you mind sharing the result of the previous command? - Did this command succeed?

python3 -m nanoowl.build_image_encoder_engine \
    data/owlvit_image_encoder_patch32.engine

Best,
John

that was the ticket.

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.