[orin] nvidia orin 上配置tensorrt | 解决虚拟环境中路径无法import的问题

版本:jetpack 5.0.1,tensorrt 8.4.1.5

概述

tensorrt会跟着jetpack的包一起安装。系统本身自带的python是3.8的版本。tensorrt的python包位于这个路径下:

cd  /usr/lib/python3.8/dist-packages/

但是由于我们是有一个虚拟环境的,所以只需要把原系统中的tensorrt组件复制过来就解决问题了。

解决方法

到系统本身的python目录下,把tensorrt-8.4.1.5.dist-infotensorrt两个文件夹拷贝到虚拟环境(我这里的名字叫做mmdeploy)中。

tensorrt-8.4.1.5.dist-info(72KB)和tensorrt(3.1MB)文件都非常小,可以放心大胆的直接复制到虚拟环境中。

cd  /usr/lib/python3.8/dist-packages/ 
sudo cp -r tensorrt-8.4.1.5.dist-info/ ~/miniforge-pypy3/envs/mmdeploy/lib/python3.8/site-packages/tensorrt-8.4.1.5.dist-info
sudo cp -r tensorrt ~/miniforge-pypy3/envs/mmdeploy/lib/python3.8/site-packages/tensorrt

测试

然后测试一下是否已经解决:
[orin] nvidia orin 上配置tensorrt | 解决虚拟环境中路径无法import的问题_第1张图片

可以看到已经能够在python环境中import了!

你可能感兴趣的:(jetson,python,linux,开发语言)