cuDNN安装

最近在用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.0-prod.tgz 

复制移动文件
>sudo cp cuda/include/cudnn.h /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/cudnn.h /usr/local/cuda/lib64/libcudnn* 


你可能感兴趣的:(Caffe,cuDNN,tensorflow)