安装完TensorRT后,导入报错
ImportError: libcudnn.so.8: cannot open shared object file: No such file or directory
上面报错信息的意思主要是liblibcudnn.so.8
这个文件找不到
-
在网上查资料后知道了这个文件是在cudnn的安装包里的,那需要先安装cudnn
-
NVIDIA 官网下载 cuDNN 8.2.1 CUDA 11.x 安装包并解压到当前目录
-
再把环境变量配置好
vim /.bashrc
,加上以下语句export LD_LIBRARY_PATH=/data/FWJ/cuda/lib64:$LD_LIBRARY_PATH