TensorRT: export failure: libcudnn.so.8: cannot open shared object file: No such file or directory
时间: 2023-11-18 07:56:53 浏览: 418
libcusparse.so.11
5星 · 资源好评率100%
这个错误提示表明在使用TensorRT进行模型导出时,缺少了名为"libcudnn.so.8"的共享对象文件,导致导出失败。cudnn是NVIDIA提供的深度学习库,TensorRT是NVIDIA的深度学习推理引擎,两者都需要依赖cudnn库。因此,要解决这个问题,需要安装cudnn库并将其路径添加到LD_LIBRARY_PATH环境变量中。具体步骤如下:
1. 下载并安装cudnn库,可以从NVIDIA官网下载对应版本的cudnn库。
2. 将cudnn库的路径添加到LD_LIBRARY_PATH环境变量中,可以通过以下命令实现:
export LD_LIBRARY_PATH=/path/to/cudnn/lib:$LD_LIBRARY_PATH
其中,/path/to/cudnn/lib是cudnn库的路径。
3. 重新运行TensorRT进行模型导出。
阅读全文