版本: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-info
和tensorrt
两个文件夹拷贝到虚拟环境(我这里的名字叫做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
测试
然后测试一下是否已经解决:
可以看到已经能够在python环境中import了!