目录
背景
通过我往期写过的教程下来,jetpack中其实已经有了tensorrt,但是我们在使用环境的过程中,使用的是虚拟环境,但是tensorrt是在主环境中的,如果要使用,要么自己再重新编译一个,但是这个是在是很麻烦,我在网上找了很多教程,都没有完整地配置出来,所以我才选择以软链接的形式直接使用主环境中的tensorrt,具体操作步骤如下:
1. 查看主环境中tensorrt的位置
直接打开一个新的终端输入以下命令
python3.6 -c "import tensorrt; print(tensorrt.__file__)"
这会在终端打印出有关tensorrt包的路径,你需要将它记住
2. 创建虚拟环境到主环境tensorrt的软链接
在终端中输入
ln -s /usr/lib/python3.8/dist-packages/tensorrt /home/x/archiconda3/envs/yolov8/lib/python3.8/site-packages/tensorrt
当然,有必要解释一下这段命令
ln -s [a] [b] 是把目录a到目录b创建符号链接
/usr/lib/python3.8/dist-packages/tensorrt ---------是在我的系统中tensorrt的目录,只要你是跟着我前面教程一套走下来的,这个目录应该是一样的,为了避免错误,这个就是上面步骤1中命令运行后得到的目录
/home/x/archiconda3/envs/yolov8/lib/python3.8/site-packages/tensorrt ----------这个目录则是我虚拟环境中的目录,我的anaconda安装在/home/x/下,虚拟环境名称为yolov8且python版本为3.8,tensorrt这个文件夹不需要自己创建
到此,你的虚拟环境中便可以在python脚本中直接使用tensorrt了
当然,你可以使用pip list查看一下是否存在,或者直接运行python脚本中import tensorrt是否报错就可验证