最近在用tensorflow跑程序,要用到cuDNN,顺便弥补下以前用caffe没安装cuDNN的偷懒
随手记录下安装方法
安装cuDNN
Theano也支持cuDNN(可选),而Tensorflow则必需要cuDNN
首先去cuDNN官网下载文件,官网下载cuDNN v5.1 Library for Linux即可,解压缩后里面其实就运行库。
cuDNN download(需要注册),下载完毕后执行
解压缩>tar xvzf cudnn-7.0-Linux-x64-v4.
复制移动文件>sudo cp cuda/include/ /usr/local/cuda/include
>sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
创建新的软连接>sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
即.h文件和运行库的软连接:
$ sudo chmod a+r /usr/local/cuda/include/ /usr/local/cuda/lib64/libcudnn*