玩转NVIDIA Jetson (26)--- 在python虚拟环境中快速使用tensorRT

python版本的tensorRT是跟随jetpack已经安装好的,但是在编译的时候只适配了python2.7和python3.6版本,也就是jetpack自带的python版本,如果需要在自己创建的conda环境中进行使用,会出现无法引入包的情况。当然可以通过重新安装tensorRT来实现,但比较麻烦,如果你的虚拟环境和原机自带的python大版本相同,例如都是python3.6,其实可以使用一个最简单的办法来应用tensorRT。

1.PYTHONPATH环境变量

pythonpath环境变量是导入包的时候自动检索的目录,默认情况下是检索python解释器所在文件目录下的包,我们只需要将这个变量多配置一个原机自带的python包目录,就可以快速使用tensorRT了。

2.切换到你所使用的虚拟环境,安装pycuda

pip install pycuda

3.添加pythonpath,注意,添加的路径是你默认可以import tensorRT的路径

export PYTHONPATH=/usr/lib/python3.6/dist-packages:$PYTHONPATH

4.测试

import tensorrt

猜你喜欢

转载自blog.csdn.net/huiyuanliyan/article/details/126691945
今日推荐