赞
踩
在执行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*
步骤二:进入虚拟环境目录下
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
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。