I’m trying to run a TensorRT Tiny YOLOv4 model on a 4GB Nano development board. I’m running the Nano headless, so should have enough memory (details in jtop output below). And, the model inference definitely works fine on my MacBook CPU. [Screenshot 2020-12-17 at 13.20.40] But, whenever I try to…

Adding the following lines to the top of the detect script (before TensorFlow imports) resolves the issue: import os os.environ['CUDA_VISIBLE_DEVICES'] = '1' Thanks to @paul.damsa for the solution!