Jetson Nano - Tensorflow no module named 'object_detection'

I installed Tensorflow using the following command:

sudo pip3 install --pre --no-cache-dir --extra-index-url tensorflow-gpu

as seen in this comment:

My objective is to train an Object Detection model. I had done this before but it was on CPU only and it took a long time. I am using the Nano to hopefully speed up the process.

I am following this guide:

And I’m at the part where I’m creating the TFRecords for training. I am using the code in this link:

but when I run it, I can’t proceed because as the code hits this line:

from object_detection.utils import dataset_util

I get:

ModuleNotFoundError: No module named 'object_detection'

But I can import tensorflow just fine.

Any ideas?

Solved it.

I followed this answer: