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