ubuntu18.04安装tensorrt
程序员文章站
2022-06-05 20:42:00
...
解压下载的文件
添加环境变量
vim ~/.bashrc
export LD_LIBRARY_PATH=/home/liguiyuan/software/TensorRT-7.0.0.11/lib:$LD_LIBRARY_PATH
source ~/.bashrc
进入解压目录 使用pip安装
TensorRT
UFF
graphsurgeon
验证安装
>>> import tensorrt
>>> tensorrt.__version__
'7.0.0.11'
错误
ImportError: libcudnn.so.7: cannot open shared object file: No such file or directory
解决办法
sudo cp TensorRT-6.01/targets/x86_64-linux-gnu/lib/lib* /usr/lib/
sudo ldconfig #立即生效