在linux服务器安装TensorRT出现以下错误
ImportError: libnvonnxparser.so.6: cannot open shared object file: No such file or directory
ImportError: libcudnn.so.7: cannot open shared object file: No such file or directory
1.解决办法:出现问题的原因要么是环境变量没有添加完成,要么是CUDNN连接建立问题。
我的是环境变量的问题
vim ~/.bashrc
- 根据自己cuda的路径改下面三行指令,不要照抄
export PATH=/home/tukrin/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/home/tukrin/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/home/tukrin/cuda
然后输入指令保存退出,不会vim看一下这个
source ~/.bashrc
让路径立即生效
- 验证TensorRT是否安装成功
python
import tensorrt
print( tensorrt.__version__)