系统版本:Ubuntu 15.04
Python版本:2.7
cuda版本:7.0 (输入nvcc --version可查看版本号)
cuDNN版本:7.0
错误原因:
目前TensorFlow不支持7.0版本的cuDNN,最高支持cuDNN 6.5 v2。
解决方法:
下载cuDNN 6.5 v2:
前往NVIDIA官网(https://developer.nvidia.com/cudnn)下载6.5版本的cuDNN。老版本可以在Archived cuDNN Releases选项中找到,下载cuDNN v2 Library for Linux而不是R1版本,文件名是cudnn-6.5-linux-x64-v2.tgz。
解压并安装:
说是安装实际上就是把解压后的文件复制到对应的includ和lib目录并链接。
复制cuDNN文件
进入解压后的文件夹,输入下面命令:
$ sudo cp cudnn.h /usr/local/include
$ sudo cp libcudnn.so /usr/local/lib
$ sudo cp libcudnn.so.6.5 /usr/local/lib
$ sudo cp libcudnn.so.6.5.48 /usr/local/lib
链接cuDNN的库文件
$ sudo ln -sf /usr/local/lib/libcudnn.so.6.5.48 /usr/local/lib/libcudnn.so.6.5
$ sudo ln -sf /usr/local/lib/libcudnn.so.6.5 /usr/local/lib/libcudnn.so
$ sudo ldconfig -v
google开源深度学习框架tensorflow环境搭配:http://www.linuxdiyf.com/linux/16153.html
ubuntu15.10源码安装tensorflow:http://www.linuxdiyf.com/linux/16089.html