- 首先查看自己是否安装了tensorrt(如果没有,就先执行下面安装)
pip install tensorrt
- 控制台输入如下代码,查看tensorrt包的路径
python3 -c "import tensorrt; print(tensorrt.__path__)"
找到自己安装包的对应路径
3. 接下来,终端运行如下代码,进行编辑
vi ~/.bashrc
- 将如下环境变量添加(请添加自己的路径奥)
export PATH=/test_for_learnning/tensorrt/:$PATH
export LD_LIBRARY_PATH=/test_for_learnning/tensorrt_libs:$LD_LIBRARY_PATH
export LIBRARY_PATH=/test_for_learnning/tensorrt/:$LIBRARY_PATH
- 保存文件
source ~/.bashrc