ImportError: libcudnn.so.8: cannot open shared object file: No such file or directory

import tensorrtTraceback (most recent call last):File "<stdin>", line 1, in <module>File "/home/zhanglu/anaconda3/lib/python3.5/site-packages/tensorrt/__init__.py", line 1, in <module&g

cv-daily

8018人浏览 · 2021-08-03 15:23:42

cv-daily · 2021-08-03 15:23:42 发布

import tensorrt
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/zhanglu/anaconda3/lib/python3.5/site-packages/tensorrt/__init__.py", line 1, in <module>
    from .tensorrt import *
ImportError: libcudnn.so.8: cannot open shared object file: No such file or directory

安装了cuda9.1，后来又装了cuda10.0，编译：

cd /usr/local/cuda-10.0/samples/1_Utilities/deviceQuery
make
sudo ./deviceQuery

软链接到cuda10.0之后。软链接：
sudo rm -rf cuda
sudo ln -s /usr/local/cuda-9.1 /usr/local/cuda

/usr/local/cuda里边找不到libcudnn.so.8。需要找到cudnn然后复制到cuda路径下。
下载：cudnn-10.2-linux-x64-v8.2.2.26.tgz
然后解压，把inculde和lib64里的文件复制到cuda中

 sudo cp include/cudnn.h /usr/local/cuda/include
sudo cp lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*