Extremely frustrating after getting it to work on my PC to train a great model, then finally getting TRT optimization to work as well. Now the final step of running a nice FP16 optimized model and I can’t even get the API installed LOL!
Thank you I found a github repo with sh files to install it on the jetson but was for protobuf 3.8 so I went and got the latest version for arm and I edited the sh files to install that one.
The install completed without error and protobuf 3.15.6 is installed into usr/local/lib/python3.6/dist-packages/protobuf-3.15.6-py3.6-linux-aarch64.egg
and there is also a usr/local/include/google/protobuf folder as well.
when I run my script now I am getting no module google.protobuf
Sorry I am not a big Linux user so if you could help me with setting the path to that folder as it is definitely in there I would appreciate it!
.py", line 18, in <module>
import tensorflow as tf
File "/usr/local/lib/python3.6/dist-packages/tensorflow/__init__.py", line 41, in <module>
from tensorflow.python.tools import module_util as _module_util
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/__init__.py", line 40, in <module>
from tensorflow.python.eager import context
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/eager/context.py", line 32, in <module>
from tensorflow.core.framework import function_pb2
File "/usr/local/lib/python3.6/dist-packages/tensorflow/core/framework/function_pb2.py", line 7, in <module>
from google.protobuf import descriptor as _descriptor
ModuleNotFoundError: No module named 'google.protobuf'
EDIT: I can import google.auth but not google.protobuf , even though if I type import google. in VSCode the auto complete finds protobuf in the list of suggestions. But it wont import if I select it and run the script. Confused!
EDIT: So it turns out in TF2 you need to delete the innit py file inside the installed protobuf subfolder google/ and then it works!