caffe 加入 cudnn编译

首先要使用cudnn v2，

ALL PLATFORMS

Extract the cuDNN archive to a directory of your choice, referred to below as <installpath>.
Then follow the platform-specific instructions as follows.

LINUX

cd <installpath>
export LD_LIBRARY_PATH=`pwd`:$LD_LIBRARY_PATH

Add <installpath> to your build and link process by adding -I<installpath> to your compile
line and -L<installpath> -lcudnn to your link line.

注意的是

在Makefile中新添的语句要在之前的前面

LDFLAGS += -L/home/research/sunmeng/software/cudnn-6.5-linux-x64-v2 -lcudnn
LDFLAGS += $(foreach librarydir,$(LIBRARY_DIRS),-L$(librarydir)) $(PKG_CONFIG) \
$(foreach library,$(LIBRARIES),-l$(library))

遇到的一个问题是把caffe编译为一个动态链接库，其他程序调用caffe.so时，也需要加载cudnn.so，则编译选项添加 -L/usr/local/cuda/lib64 -L/home/xxxxx/cudnn-6.5-linux-x64-v2/

一直提示是undefined reference to `cudnnGetConvolutionForwardWorkspaceSize' 之类的错误，这是因为在动态链接库中找不到相应的函数实现

这是因为之前在 /usr/local/cuda/lib64目录下拷贝了一个老版本的cudnn.so, 而我们想要使用的是新版本的/home/xxxxx/cudnn-6.5-linux-x64-v2，g++先在lib64找，找到了就不在后面的目录找了，解决这个问题很简单，就是把两个目录的顺序调换即可

///

程序在运行时就要需要加载动态链接库了，原理参考

http://blog.sina.com.cn/s/blog_484102dd0101114f.html

所以安装共享库后要注意共享库路径设置问题, 如下:

1) 如果共享库文件安装到了/lib或/usr/lib目录下, 那么需执行一下ldconfig命令

ldconfig命令的用途, 主要是在默认搜寻目录(/lib和/usr/lib)以及动态库配置文件/etc/ld.so.conf内所列的目录下, 搜索出可共享的动态链接库(格式如lib*.so*), 进而创建出动态装入程序(ld.so)所需的连接和缓存文件. 缓存文件默认为/etc/ld.so.cache, 此文件保存已排好序的动态链接库名字列表.

2) 如果共享库文件安装到了/usr/local/lib(很多开源的共享库都会安装到该目录下)或其它"非/lib或/usr/lib"目录下, 那么在执行ldconfig命令前, 还要把新共享库目录加入到共享库配置文件/etc/ld.so.conf中, 如下:

# cat /etc/ld.so.conf
include ld.so.conf.d/*.conf
# echo "/usr/local/lib" >> /etc/ld.so.conf
# ldconfig

3) 如果共享库文件安装到了其它"非/lib或/usr/lib" 目录下, 但是又不想在/etc/ld.so.conf中加路径(或者是没有权限加路径). 那可以export一个全局变量LD_LIBRARY_PATH, 然后运行程序的时候就会去这个目录中找共享库.

LD_LIBRARY_PATH的意思是告诉loader在哪些目录中可以找到共享库. 可以设置多个搜索目录, 这些目录之间用冒号分隔开. 比如安装了一个mysql到/usr/local/mysql目录下, 其中有一大堆库文件在/usr/local/mysql/lib下面, 则可以在.bashrc或.bash_profile或shell里加入以下语句即可:

export LD_LIBRARY_PATH=/usr/local/mysql/lib:$LD_LIBRARY_PATH

一般来讲这只是一种临时的解决方案, 在没有权限或临时需要的时候使用.

当然也可以将动态链接库的地址写入到可执行文件中去，这样在运行程序的时候就不用考虑动态链接库的地址了，注意，最好是绝对路径，就是在编译中添加

-Wl,-rpath,/home/xxxx/cudnn-6.5-linux-x64-v2:/usr/local/cuda/lib64:

caffe 加入 cudnn编译相关推荐

Ubuntu 1６.04 LTS, 64bit,cuda ８, Caffe环境配置编译和安装
1 GPU设备 1.1 本人使用的GPU设备名称: 七彩虹GTX980Ti显卡iGame980Ti烈焰战神X 6G 1．2 硬件问题 Laboratory Tested Hardware: Berk ...
Ubuntu 14.04 LTS, 64bit, cuda 7, Caffe环境配置编译和安装
1 GPU设备 1.1 本人使用的GPU设备名称: 七彩虹GTX980Ti显卡iGame980Ti烈焰战神X 6G 1．2 硬件问题 Laboratory Tested Hardware: Berk ...
./include/caffe/util/cudnn.hpp: error: too few arguments to function ‘SetPooling2dDescriptor
https://blog.csdn.net/tianrolin/article/details/71246472 这是由于所使用的cuDNN版本不一致的导致的,作者配置环境是cuDNN 4.0,但是5 ...
caffe cmake matlab,编译caffe时候抛出的错误
编译caffe时候抛出的错误我在安装ubuntu14.04上安装了cuda7.5,cudnn4.0等,使用起来都挺好的,没什么问题,最近,把21天实战caffe上面第五天的以来库都装完之后,准备装m ...
caffe SSD 代码编译运行流程及问题解决
caffe SSD 代码编译运行流程及问题解决该文基于以下代码: https://github.com/weiliu89/caffe/tree/ssd down下来后,进入目录 -rw-rw-r-- ...
Deep Learning for Computer Vision with Caffe and cuDNN
转载自:Deep Learning for Computer Vision with Caffe and cuDNN | Parallel Forall http://devblogs.nvidia. ...
caffe 使用cudnn 加速报错
./include/caffe/util/cudnn.hpp: In function 'void caffe::cudnn::createPoolingDesc(cudnnPoolingStruct ...
caffe安装，编译（包括CUDA和cuDNN的安装），并训练，测试自己的数据（caffe使用教程）
caffe是一个非常清晰且高效的深度学习框架,目前有着不少的用户,也渐渐的形成了自己的社区,社区上可以讨论相关的问题. 我从开始看深度学习的相关内容到能够用caffe训练测试自己的数据,看了不少网站, ...
Ubuntu14.04 + Matlab2014a + caffe + cuda + cudnn环境搭建
为什么80%的码农都做不了架构师?>>> 安装Caffe环境前,请查好各个软件版本的相互匹配资料,确定好所有软件版本后,再开始安装. 一.安装gcc4.6 Ubuntu14.0 ...

caffe 加入 cudnn编译

caffe 加入 cudnn编译相关推荐

最新文章

热门文章