Ubuntu18+CUDA10+TensorRT中pycharm ImportError: libcudnn.so.6缺少,类似错误解决办法
正常安装cuda+cudnn+tensorrt,并且通过vim ~/.bashrc设置了环境变量,在命令行使用python直接import也没问题,但是在pycharm中import报错,缺少so文件
方法一
将TensorRT中的链接文件.so文件复制到/usr/lib/文件夹中,比如
1) ImportError: libnvinfer.so.6: cannot open shared object file: Nosuch file or directory
2)ImportError: libnvonnxparser.so.6: cannot open shared object file:No such file or directory
解决办法:
1) sudo cp TensorRT-6.01/targets/x86_64-linux-gnu/lib/libnvinfer.so.6 /usr/lib/
2)sudo cp TensorRT-6.01/targets/x86_64-linux-gnu/lib/libnvonnxparser.so.6 /usr/lib/
方法二
在pycharm菜单Run->Edit configurations 中,手动设置Environment variables,添加LD_LIBRARY_PATH的内容:
/usr/local/TensorRT-6.0.1.5/lib
,重启pycharm即可解决问题。