运行TensorFlow-GPU的时候代码报错:
Could not load dynamic library 'libcudnn.so.7'; dlerror: libcudnn.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/cuda-8.0/lib64:/usr/local/cuda/lib64
Could not load dynamic library 'libcudart.so.10.0'; dlerror: libcudart.so.10.0: cannot open ...
...
如下图显示:
提示找不到cuda、cudnn动态库文件。
原因有2种:1.cuda、cudnn版本与显卡驱动版本对应错误。2. 安装了正确的版本,却没有软链接到系统环境。
提供一个无需繁琐的重装cuda或设置系统环境变量的解决方法。
解决办法:
以安装TensorFlow1.15为例,需要配套安装cuda10.0 + cudnn7.6.5,仅需要
conda install tensorflow-gpu=1.15
注意一定要conda install 而非pip install !!!
完美解决报错问题