jetson 系列安装完jetpack/已安装 tensorrt 在虚拟环境中仍然报 no module named tensorrt ＞＞在虚拟环境建立软连接

uptnv

已于 2022-03-02 03:41:48 修改

阅读量8.2k

点赞数 14

分类专栏： ubuntu 文章标签： c++ linux

于 2022-03-02 02:17:18 首次发布

本文链接：https://blog.csdn.net/qq_34370204/article/details/123220598

版权

ubuntu 专栏收录该内容

7 篇文章

订阅专栏

当使用虚拟环境时，需要考虑

cv2 和 tensorrt 是一样的

更新：cv2 应该是对的。
tensorrt的处理是错的，在使用 trt时很多方法都找不到。正确的加入虚拟环境的方法：
https://www.bojankomazec.com/2019/12/how-to-install-tensorrt-python-package.html

cv2安装 – 与虚拟环境链接

为了使用TensorRT完成之后的识别功能，也为了推理的更快，我们当然必须要安装opencv

NVIDIA Jetson AGX Xavier学习笔记3——环境配置(pytorch、torchvision、cv2)

进行NVIDIA Jetson AGX Xavier开发组件刷机时已经自动安装了OpenCV，只需要将python虚拟环境链接到OpenCV库即可。

安装完成后，在虚拟环境中执行下列指令以确保python能正确调用cv2

python               //进入python
import cv2
cv2.__version__    //若安装成功且能正常调用，此处能输出安装的从v版本
quit()

实测在当前环境中找不到 cv2 虚拟环境可以运行。使用base下的 python2 可以运行找到。

首先在终端执行以下指令查找编译好的cv2库文件的路径。

sudo find / -iname "*cv2*"
// 得到路径 /usr/lib/python3.6/dist-packages/cv2/python-3.6/cv2.cython-36m-aarch64-linux-gnu.so

之后进入<虚拟环境> 的 site-packages文件夹下,并链接到查找到的cv2库文件路径即可。

cd /home/miniforge3/envs/<virtual>/lib/python3.6/site-packages  # 其中virtual是我建立的虚拟环境名
ln -s /usr/lib/python3.6/dist-packages/cv2/python-3.6/cv2.cython-36m-aarch64-linux-gnu.so cv2.so

之后在这个环境中就可以正常验证了：

python               //进入python
import cv2
cv2.__version__    //若安装成功且能正常调用，此处能输出安装的从v版本
quit()               //退出python

对于 tensorrt 也是

(base) agxxavier@agxxavier-desktop:~$ sudo find / -iname "*tensorrt*"
/usr/lib/python3.6/dist-packages/tensorrt
/usr/lib/python3.6/dist-packages/tensorrt/tensorrt.so   // 这个路径
/usr/lib/python3.6/dist-packages/tensorrt-8.0.1.6.dist-info

注：从文件夹进去竟然看不到，，从终端进去ls可以看到…为什么

系统bug 关闭文件管理器重进能看到

之后进入<虚拟环境> 的 site-packages文件夹下,并链接到查找到的tensorrt库文件路径即可。

(deepNN) agxxavier@agxxavier-desktop:~/miniforge3/envs/deepNN/lib/python3.6/site-packages$ ln -s /usr/lib/python3.6/dist-packages/tensorrt/tensorrt.so tensorrt.so