在运行GPU版的tensorflow时,遇到ImportError: libcudnn.so.5: cannot open shared object file: No such file or directory。我安装的tensorflow-gpu=1.2
查看linux系统安装的CUDA,在目录/usr/local/cuda-8.0/lib64下:
1 | libcublas_device.a libcufft.so.8.0 libcusolver.so libnppicc.so libnppim.so libnppitc.so.8.0.61 libnvrtc.so |
确实没有libcudnn.so.5
tensorflow要和安装的CUDA对应,CUDA是NVIDIA的编程语言平台,想使用GPU就必须要使用cuda。
- 在官网可以查看适合自己显卡的驱动
- 安装cuda(https://developer.nvidia.com/cuda-downloads )
- 安装cuDNN,cuDNN是GPU加速计算深层神经网络的库。 (https://developer.nvidia.com/rdp/cudnn-download)
- 安装tensorflow,可以在清华镜像网址上找对应版https://mirror.tuna.tsinghua.edu.cn/help/tensorflow/>
注:cudnn6支持的是tensorflow1.3及以上
查看CUDA版本:
1 | cat /usr/local/cuda/version.txt |
1 | CUDA Version 8.0.61 |
查看cudnn版本
1 | cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 |
- *