agx orin设备使用trt进行yolo算法加速
1.加载RT加速
在之前 首先需要安装Jetson设备的jetpack Torch 。
我们在最开始刷机的时候,会将RT一起装到agx里面
1.1.查看一下是否安装上
ls -d /usr/lib/python3.10/dist-packages/tensorrt*
1.2.软链接映射
请直接在终端(确保在 (conda) 环境下)复制并运行下面这一段指令,就可以搞定了:
# 强制建立软链接 (覆盖旧的或错误的链接)
sudo ln -sf /usr/lib/python3.10/dist-packages/tensorrt* $CONDA_PREFIX/lib/python3.10/site-packages/这行命令会把你列出的 tensorrt, tensorrt_dispatch, tensorrt_lean 全部映射过去。
配置环境变量
因为是 TensorRT 10.x,对动态库的依赖更严格,如果不加这个,Python 可能会报错找不到库文件。
# 1. 将系统库路径写入当前环境的配置文件
mkdir -p $CONDA_PREFIX/etc/conda/activate.d echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/aarch64-linux-gnu/' >> $CONDA_PREFIX/etc/conda/ac