ModuleNotFoundError: No module named 'graphsurgeon'

Please provide complete information as applicable to your setup.

• Hardware Platform (Jetson / GPU) NVIDIA RTX A3000
• DeepStream Version 6.2

• TensorRT Version
**• NVIDIA GPU Driver Version (valid for GPU only)****• 525.105

I am trying to run this model tracker_NvDeepSORT
graphsurgeon-tf and onnx-graphsurgeon is already installed.
Still I am getting this error
/opt/nvidia/deepstream/deepstream-6.2/sources/tracker_NvDeepSORT$ python3
Traceback (most recent call last):
File “”, line 17, in
import graphsurgeon as gs
ModuleNotFoundError: No module named ‘graphsurgeon’

Are you using DeepStream docker? Which docker are you using?

Please refer here if you want to convert DeepSORT model: Deepstream Tracker FAQ - #5 by kesong

No I am not using Docker. I have installed DeepStream 6.2 on my local Ubuntu device.

I manually installed graphsurgeon-tf and onnx-graphsurgeon using .whl file available in TensorRT tar file.
Now that error is gone.
But I am getting this tensorflow error. I followed this to install tensorflow Deepstream Tracker FAQ - #5 by kesong
I did pip3 install tensorflow==2.11.0 PyYAML

ModuleNotFoundError: No module named ‘tensorflow’

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File “”, line 17, in
import graphsurgeon as gs
File “/usr/lib/python3.8/dist-packages/graphsurgeon/”, line 51, in
raise ImportError(“”"ERROR: Failed to import module ({})
ImportError: ERROR: Failed to import module (No module named ‘tensorflow’)

There is no update from you for a period, assuming this is not an issue anymore. Hence we are closing this topic. If need further support, please open a new one. Thanks

Seems it is your setup issue. Can you have a try with DeepStream docker?

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