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*

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

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
在这里插入图片描述

  • 4
    点赞
  • 36
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值