当前位置:   article > 正文

jetson xavier在使用tensorRT对yolov5加速时,报错ModuleNotFoundError: No module named ‘tensorrt‘_modulenotfounderror: no module named 'tensorrt

modulenotfounderror: no module named 'tensorrt

在执行python yolov5_trt.py

先是报错ModuleNotFoundError: No module named 'pycuda';
解决办法:pip install pycuda

之后报错:ModuleNotFoundError: No module named 'tensorrt'
错误示范:pip install tensorrt 不可行

解决办法:前提是你的tensorrt已经安装了,但是虚拟环境里没有,此方法才可行
步骤一:查看tensorrt库在那里:

sudo find / -name tensorrt*
  • 1

在这里插入图片描述
步骤二:进入虚拟环境目录下

cd /home/nvidia/mambaforge/envs/yolov5/lib/python3.6/site-packages
  • 1

设置软连接

 ln -s /usr/lib/python3.6/dist-packages/tensorrt
  • 1

这个不行的话设置:

 ln -s /usr/lib/python3.6/dist-packages/tensorrt/tensorrt.so
  • 1

在这里插入图片描述
再次执行:python yolov5_trt.py
在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Cpp五条/article/detail/129831
推荐阅读
相关标签
  

闽ICP备14008679号