Closed bmox closed 11 months ago
This has nothing to do with this repo. Your tensorrt lib is not linked correctly.
See as references: https://github.com/NVIDIA/TensorRT/issues/2596 https://github.com/NVIDIA/TensorRT/issues/1817
For me I had to do this:
python3 -c "import tensorrt; print(tensorrt.__path__)"
Output
['/usr/local/lib/python3.10/dist-packages/tensorrt']
The lib for me was actually under (so scoop around a bit in your package folders): /usr/local/lib/python3.10/dist-packages/tensorrt_libs
echo $LD_LIBRARY_PATH
/usr/local/nvidia/lib:/usr/local/nvidia/lib64
As you can see the lib was not in my path. You can add it via:
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/lib/python3.10/dist-packages/tensorrt_libs"
afterwords it should look like this:
echo $LD_LIBRARY_PATH
/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/lib/python3.10/dist-packages/tensorrt_libs
Hope it helps
I'm running this on Google Colab