jetson xavier在使用tensorRT对yolov5加速时,报错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*

jetson xavier在使用tensorRT对yolov5加速时,报错ModuleNotFoundError: No module named ‘tensorrt‘_第1张图片
步骤二:进入虚拟环境目录下

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

设置软连接

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

这个不行的话设置:

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

在这里插入图片描述
再次执行:python yolov5_trt.py
jetson xavier在使用tensorRT对yolov5加速时,报错ModuleNotFoundError: No module named ‘tensorrt‘_第2张图片

你可能感兴趣的:(xavier,目标检测)