I installed Tensorflow using the following command:
sudo pip3 install --pre --no-cache-dir --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu
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: https://towardsdatascience.com/creating-your-own-object-detector-ad69dda69c85
And I’m at the part where I’m creating the TFRecords for training. I am using the code in this link: https://github.com/datitran/raccoon_dataset/blob/master/generate_tfrecord.py
but when I run it, I can’t proceed because as the code hits this line:
from object_detection.utils import dataset_util
ModuleNotFoundError: No module named 'object_detection'
But I can import tensorflow just fine.